From 61d3493f87e8b04fc5eb29d5bed1eafcc52db6bc Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Mon, 21 Mar 2022 11:12:55 -0700 Subject: [PATCH] Fix hf-hub handling when hf-hub is config source --- timm/models/helpers.py | 1 + 1 file changed, 1 insertion(+) diff --git a/timm/models/helpers.py b/timm/models/helpers.py index 9afe3607..eda09680 100644 --- 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: