ZeqiangLai commited on
Commit
dfd9bf4
·
verified ·
1 Parent(s): 4becb3e

Update hy3dgen/texgen/pipelines.py

Browse files
Files changed (1) hide show
  1. hy3dgen/texgen/pipelines.py +5 -1
hy3dgen/texgen/pipelines.py CHANGED
@@ -71,9 +71,13 @@ class Hunyuan3DPaintPipeline:
71
  try:
72
  import huggingface_hub
73
  # download from huggingface
74
- model_path = huggingface_hub.snapshot_download(repo_id=original_model_path)
 
 
 
75
  delight_model_path = os.path.join(model_path, 'hunyuan3d-delight-v2-0')
76
  multiview_model_path = os.path.join(model_path, 'hunyuan3d-paint-v2-0')
 
77
  return cls(Hunyuan3DTexGenConfig(delight_model_path, multiview_model_path))
78
  except ImportError:
79
  logger.warning(
 
71
  try:
72
  import huggingface_hub
73
  # download from huggingface
74
+ model_path = huggingface_hub.snapshot_download(repo_id=original_model_path,
75
+ allow_patterns=["hunyuan3d-delight-v2-0/*"])
76
+ model_path = huggingface_hub.snapshot_download(repo_id=original_model_path,
77
+ allow_patterns=["hunyuan3d-paint-v2-0/*"])
78
  delight_model_path = os.path.join(model_path, 'hunyuan3d-delight-v2-0')
79
  multiview_model_path = os.path.join(model_path, 'hunyuan3d-paint-v2-0')
80
+
81
  return cls(Hunyuan3DTexGenConfig(delight_model_path, multiview_model_path))
82
  except ImportError:
83
  logger.warning(