Make huggingface_hub necessary if it's the only source for a pretrain…

…ed weight
huggingface · Sep 23, 2022 · a383ef9 · a383ef9
1 parent d199f66
commit a383ef9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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