From 8fa86a28a8eb5ecb5d62b9cf5804c2fdd22717b0 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Mon, 1 May 2023 10:24:08 -0700 Subject: [PATCH] Add datacomp L/14 (79.2 zs) image tower 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 cf88528cf4..75697f972c 100644 --- a/timm/models/vision_transformer.py +++ b/timm/models/vision_transformer.py @@ -1197,6 +1197,10 @@ def _cfg(url='', **kwargs): hf_hub_id='laion/CLIP-ViT-L-14-laion2B-s32B-b82K', hf_hub_filename='open_clip_pytorch_model.bin', mean=IMAGENET_INCEPTION_MEAN, std=IMAGENET_INCEPTION_STD, crop_pct=1.0, num_classes=768), + 'vit_large_patch14_clip_224.datacompxl': _cfg( + hf_hub_id='laion/CLIP-ViT-L-14-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=768), 'vit_huge_patch14_clip_224.laion2b': _cfg( hf_hub_id='laion/CLIP-ViT-H-14-laion2B-s32B-b79K', hf_hub_filename='open_clip_pytorch_model.bin',