From c5d3ee47f3a21b42bbdfc22bea092fcc0e7b897f Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Tue, 16 May 2023 11:27:20 -0700 Subject: [PATCH] Add B/16 datacompxl CLIP weights --- timm/models/vision_transformer.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/timm/models/vision_transformer.py b/timm/models/vision_transformer.py index e0f0bdfd..0104e007 100644 --- a/timm/models/vision_transformer.py +++ b/timm/models/vision_transformer.py @@ -1293,6 +1293,10 @@ default_cfgs = generate_default_cfgs({ hf_hub_id='laion/CLIP-ViT-B-16-laion2B-s34B-b88K', hf_hub_filename='open_clip_pytorch_model.bin', mean=OPENAI_CLIP_MEAN, std=OPENAI_CLIP_STD, crop_pct=1.0, num_classes=512), + 'vit_base_patch16_clip_224.datacompxl': _cfg( + hf_hub_id='laion/CLIP-ViT-B-16-DataComp.XL-s13B-b90K', + hf_hub_filename='open_clip_pytorch_model.bin', + mean=OPENAI_CLIP_MEAN, std=OPENAI_CLIP_STD, crop_pct=1.0, num_classes=512), 'vit_large_patch14_clip_224.laion2b': _cfg( hf_hub_id='laion/CLIP-ViT-L-14-laion2B-s32B-b82K', hf_hub_filename='open_clip_pytorch_model.bin',