murcherful commited on
Commit
a9923b4
·
verified ·
1 Parent(s): bb0cc79

Update P3-SAM/model.py

Browse files
Files changed (1) hide show
  1. 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.ckpt", 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
 
 
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