Make huggingface_hub necessary if it's the only source for a pretrained weight

2 years ago · a383ef99f5
parent d199f6651d
commit a383ef99f5
1 changed files with 1 additions and 1 deletions
--- a/timm/models/helpers.py
+++ b/timm/models/helpers.py
@ -134,7 +134,7 @@ def _resolve_pretrained_source(pretrained_cfg):
        elif pretrained_url:
            load_from = 'url'
            pretrained_loc = pretrained_url
-        elif hf_hub_id and has_hf_hub(necessary=False):
+        elif hf_hub_id and has_hf_hub(necessary=True):
            # hf-hub available as alternate weight source in default_cfg
            load_from = 'hf-hub'
            pretrained_loc = hf_hub_id