Fix hf-hub handling when hf-hub is config source

3 years ago · 61d3493f87
parent 5f47518f27
commit 61d3493f87
1 changed files with 1 additions and 0 deletions
--- a/timm/models/helpers.py
+++ b/timm/models/helpers.py
@ -125,6 +125,7 @@ def _resolve_pretrained_source(pretrained_cfg):
        # hf-hub specified as source via model identifier
        load_from = 'hf-hub'
        assert hf_hub_id
+        pretrained_loc = hf_hub_id
    else:
        # default source == timm or unspecified
        if pretrained_file: