ZeqiangLai commited on
Commit
acaa008
·
verified ·
1 Parent(s): 80a81f4

Update hy3dgen/texgen/pipelines.py

Browse files
Files changed (1) hide show
  1. hy3dgen/texgen/pipelines.py +4 -1
hy3dgen/texgen/pipelines.py CHANGED
@@ -62,7 +62,10 @@ class Hunyuan3DPaintPipeline:
62
  try:
63
  import huggingface_hub
64
  # download from huggingface
65
- model_path = huggingface_hub.snapshot_download(repo_id=original_model_path)
 
 
 
66
  delight_model_path = os.path.join(model_path, 'hunyuan3d-delight-v2-0')
67
  multiview_model_path = os.path.join(model_path, 'hunyuan3d-paint-v2-0')
68
  return cls(Hunyuan3DTexGenConfig(delight_model_path, multiview_model_path))
 
62
  try:
63
  import huggingface_hub
64
  # download from huggingface
65
+ model_path = huggingface_hub.snapshot_download(repo_id=original_model_path,
66
+ allow_patterns=["hunyuan3d-delight-v2-0/*"])
67
+ model_path = huggingface_hub.snapshot_download(repo_id=original_model_path,
68
+ allow_patterns=["hunyuan3d-paint-v2-0/*"])
69
  delight_model_path = os.path.join(model_path, 'hunyuan3d-delight-v2-0')
70
  multiview_model_path = os.path.join(model_path, 'hunyuan3d-paint-v2-0')
71
  return cls(Hunyuan3DTexGenConfig(delight_model_path, multiview_model_path))