murcherful commited on
Commit
454fc3e
·
1 Parent(s): c674d5b

change cache path

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, 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.ckpt", local_dir='cache/P3-SAM/')
122
  print(f'download model from huggingface to: {ckpt_path}')
123
  state_dict = torch.load(ckpt_path, map_location="cpu")["state_dict"]
124