423A35C7/sam2
1
0
Fork 0
mirror of https://github.com/facebookresearch/sam2.git synced 2025-11-04 19:42:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

Do not load config from the hub

Browse Source
This commit is contained in:
Niels 2024-08-03 14:45:20 +02:00
parent 3af4e82263
commit 0c28c630c2
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
sam2/build_sam.py
View File

@ -87,9 +87,9 @@ def build_sam2_hf(model_id, **kwargs):
"facebook/sam2-hiera-large": ("sam2_hiera_l.yaml", "sam2_hiera_large.pt"), "facebook/sam2-hiera-large": ("sam2_hiera_l.yaml", "sam2_hiera_large.pt"),
} }
config_name, checkpoint_name = model_id_to_filenames[model_id] config_name, checkpoint_name = model_id_to_filenames[model_id]
config_file = hf_hub_download(repo_id=model_id, filename=config_name) # config_file = hf_hub_download(repo_id=model_id, filename=config_name)
ckpt_path = hf_hub_download(repo_id=model_id, filename=checkpoint_name) ckpt_path = hf_hub_download(repo_id=model_id, filename=checkpoint_name)
return build_sam2_video_predictor(config_file=config_file, ckpt_path=ckpt_path, **kwargs) return build_sam2_video_predictor(config_file=config_name, ckpt_path=ckpt_path, **kwargs)
def build_sam2_video_predictor_hf(model_id, **kwargs): def build_sam2_video_predictor_hf(model_id, **kwargs):