Skip to content

Commit b2b6285

Browse files
committed
Add two more FT clip weights
1 parent 5895056 commit b2b6285

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

timm/models/vision_transformer.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -816,7 +816,7 @@ def _cfg(url='', **kwargs):
816816
hf_hub_id='timm/vit_base_patch16_clip_224.laion2b_ft_in12k_in1k',
817817
mean=OPENAI_CLIP_MEAN, std=OPENAI_CLIP_STD, crop_pct=1.0),
818818
'vit_base_patch16_clip_384.laion2b_ft_in12k_in1k': _cfg(
819-
#hf_hub_id='timm/vit_base_patch16_clip_384.laion2b_ft_in12k_in1k',
819+
hf_hub_id='timm/vit_base_patch16_clip_384.laion2b_ft_in12k_in1k',
820820
mean=OPENAI_CLIP_MEAN, std=OPENAI_CLIP_STD, crop_pct=1.0, input_size=(3, 384, 384)),
821821
'vit_large_patch14_clip_224.laion2b_ft_in12k_in1k': _cfg(
822822
hf_hub_id='timm/vit_large_patch14_clip_224.laion2b_ft_in12k_in1k',
@@ -870,6 +870,9 @@ def _cfg(url='', **kwargs):
870870
'vit_base_patch32_clip_224.openai_ft_in12k_in1k': _cfg(
871871
#hf_hub_id='timm/vit_base_patch32_clip_224.openai_ft_in12k_in1k',
872872
mean=OPENAI_CLIP_MEAN, std=OPENAI_CLIP_STD),
873+
'vit_base_patch32_clip_384.openai_ft_in12k_in1k': _cfg(
874+
hf_hub_id='timm/vit_base_patch32_clip_384.openai_ft_in12k_in1k',
875+
mean=OPENAI_CLIP_MEAN, std=OPENAI_CLIP_STD, crop_pct=1.0, input_size=(3, 384, 384)),
873876
'vit_base_patch16_clip_224.openai_ft_in12k_in1k': _cfg(
874877
#hf_hub_id='timm/vit_base_patch16_clip_224.openai_ft_in12k_in1k',
875878
mean=OPENAI_CLIP_MEAN, std=OPENAI_CLIP_STD),

0 commit comments

Comments
 (0)