Spaces:
Build error
Build error
Update P3-SAM/model.py
Browse files- P3-SAM/model.py +1 -1
P3-SAM/model.py
CHANGED
|
@@ -118,7 +118,7 @@ def load_state_dict(self,
|
|
| 118 |
# download from huggingface
|
| 119 |
print(f'trying to download model from huggingface...')
|
| 120 |
from huggingface_hub import hf_hub_download
|
| 121 |
-
ckpt_path = hf_hub_download(repo_id="tencent/Hunyuan3D-Part", filename="p3sam.
|
| 122 |
print(f'download model from huggingface to: {ckpt_path}')
|
| 123 |
state_dict = torch.load(ckpt_path, weights_only=False, map_location="cpu")["state_dict"]
|
| 124 |
|
|
|
|
| 118 |
# download from huggingface
|
| 119 |
print(f'trying to download model from huggingface...')
|
| 120 |
from huggingface_hub import hf_hub_download
|
| 121 |
+
ckpt_path = hf_hub_download(repo_id="tencent/Hunyuan3D-Part", filename="p3sam.pt", local_dir='cache/P3-SAM/')
|
| 122 |
print(f'download model from huggingface to: {ckpt_path}')
|
| 123 |
state_dict = torch.load(ckpt_path, weights_only=False, map_location="cpu")["state_dict"]
|
| 124 |
|