diff --git a/timm/models/vision_transformer.py b/timm/models/vision_transformer.py index 2c47c1f00d..cf88528cf4 100644 --- a/timm/models/vision_transformer.py +++ b/timm/models/vision_transformer.py @@ -1019,10 +1019,10 @@ def _cfg(url='', **kwargs): custom_load=True, num_classes=21843), # SAM trained models (https://arxiv.org/abs/2106.01548) - 'vit_base_patch32_224.sam': _cfg( + 'vit_base_patch32_224.sam_in1k': _cfg( url='https://storage.googleapis.com/vit_models/sam/ViT-B_32.npz', custom_load=True, hf_hub_id='timm/'), - 'vit_base_patch16_224.sam': _cfg( + 'vit_base_patch16_224.sam_in1k': _cfg( url='https://storage.googleapis.com/vit_models/sam/ViT-B_16.npz', custom_load=True, hf_hub_id='timm/'), diff --git a/timm/version.py b/timm/version.py index ff9a92e259..b7b0e31206 100644 --- a/timm/version.py +++ b/timm/version.py @@ -1 +1 @@ -__version__ = '0.8.20dev0' +__version__ = '0.8.21dev0'