KawasakiAkasei/Hunyuan3D_2.1_Low_VRAM
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
Huiwenshi
2025-06-14 14:53:27 +08:00
parent dd93e7ce4e
commit d575f308d0
1 changed files with 11 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
demo.py
View File

@@ -18,17 +18,17 @@ except Exception as e:
print(f"Warning: Failed to apply torchvision fix: {e}")
# shape
#model_path = 'tencent/Hunyuan3D-2.1'
#pipeline_shapegen = Hunyuan3DDiTFlowMatchingPipeline.from_pretrained(model_path)
##
#image_path = 'assets/demo.png'
#image = Image.open(image_path).convert("RGBA")
#if image.mode == 'RGB':
# rembg = BackgroundRemover()
# image = rembg(image)
#
#mesh = pipeline_shapegen(image=image)[0]
#mesh.export('demo.glb')
model_path = 'tencent/Hunyuan3D-2.1'
pipeline_shapegen = Hunyuan3DDiTFlowMatchingPipeline.from_pretrained(model_path)
image_path = 'assets/demo.png'
image = Image.open(image_path).convert("RGBA")
if image.mode == 'RGB':
rembg = BackgroundRemover()
image = rembg(image)
mesh = pipeline_shapegen(image=image)[0]
mesh.export('demo.glb')
# paint
max_num_view = 6 # can be 6 to 9