From 4d9c3ae2fb7cc4739ec57d4c06254d2ffc7e2c89 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Sat, 18 Feb 2023 16:34:03 -0800 Subject: [PATCH] Add laion2b 320x320 ConvNeXt-Large CLIP weights --- timm/models/convnext.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/timm/models/convnext.py b/timm/models/convnext.py index ce7b5a27..a3b29953 100644 --- a/timm/models/convnext.py +++ b/timm/models/convnext.py @@ -773,6 +773,16 @@ default_cfgs = generate_default_cfgs({ hf_hub_filename='open_clip_pytorch_model.bin', mean=OPENAI_CLIP_MEAN, std=OPENAI_CLIP_STD, input_size=(3, 256, 256), pool_size=(8, 8), crop_pct=1.0, num_classes=768), + 'convnext_large_mlp.clip_laion2b_ft_320': _cfg( + hf_hub_id='laion/CLIP-convnext_large_d_320.laion2B-s29B-b131K-ft', + hf_hub_filename='open_clip_pytorch_model.bin', + mean=OPENAI_CLIP_MEAN, std=OPENAI_CLIP_STD, + input_size=(3, 320, 320), pool_size=(10, 10), crop_pct=1.0, num_classes=768), + 'convnext_large_mlp.clip_laion2b_ft_soup_320': _cfg( + hf_hub_id='laion/CLIP-convnext_large_d_320.laion2B-s29B-b131K-ft-soup', + hf_hub_filename='open_clip_pytorch_model.bin', + mean=OPENAI_CLIP_MEAN, std=OPENAI_CLIP_STD, + input_size=(3, 320, 320), pool_size=(10, 10), crop_pct=1.0, num_classes=768), })