mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Ross Wightman 2022-09-23 13:54:21 -07:00
parent d199f6651d
commit a383ef99f5
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
timm/models/helpers.py
View File

@ -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