From 0c28c630c20edff1fda3b4082378823bcd5720aa Mon Sep 17 00:00:00 2001 From: Niels Date: Sat, 3 Aug 2024 14:45:20 +0200 Subject: [PATCH] Do not load config from the hub --- sam2/build_sam.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sam2/build_sam.py b/sam2/build_sam.py index eb07ca1..00f9dcf 100644 --- a/sam2/build_sam.py +++ b/sam2/build_sam.py @@ -87,9 +87,9 @@ def build_sam2_hf(model_id, **kwargs): "facebook/sam2-hiera-large": ("sam2_hiera_l.yaml", "sam2_hiera_large.pt"), } 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) - 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):