From da2de0de9538e124246b44958414f60403d2b29f Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Wed, 9 Nov 2022 15:27:23 -0800 Subject: [PATCH] CLIP B16 12k weights added --- timm/models/vision_transformer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/timm/models/vision_transformer.py b/timm/models/vision_transformer.py index 2066581a..2db69f07 100644 --- a/timm/models/vision_transformer.py +++ b/timm/models/vision_transformer.py @@ -835,7 +835,7 @@ default_cfgs = generate_defaults({ hf_hub_id='timm/vit_base_patch32_clip_224.laion2b_ft_in12k', mean=OPENAI_CLIP_MEAN, std=OPENAI_CLIP_STD, num_classes=11821), 'vit_base_patch16_clip_224.laion2b_ft_in12k': _cfg( - #hf_hub_id='timm/vit_base_patch16_clip_224.laion2b_ft_in12k', + hf_hub_id='timm/vit_base_patch16_clip_224.laion2b_ft_in12k', mean=OPENAI_CLIP_MEAN, std=OPENAI_CLIP_STD, num_classes=11821), 'vit_large_patch14_clip_224.laion2b_ft_in12k': _cfg( hf_hub_id='timm/vit_large_patch14_clip_224.laion2b_ft_in12k',