Skip to content

Commit 4d9c3ae

Browse files
committed
Add laion2b 320x320 ConvNeXt-Large CLIP weights
1 parent d0b45c9 commit 4d9c3ae

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

timm/models/convnext.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -773,6 +773,16 @@ def _cfgv2(url='', **kwargs):
773773
hf_hub_filename='open_clip_pytorch_model.bin',
774774
mean=OPENAI_CLIP_MEAN, std=OPENAI_CLIP_STD,
775775
input_size=(3, 256, 256), pool_size=(8, 8), crop_pct=1.0, num_classes=768),
776+
'convnext_large_mlp.clip_laion2b_ft_320': _cfg(
777+
hf_hub_id='laion/CLIP-convnext_large_d_320.laion2B-s29B-b131K-ft',
778+
hf_hub_filename='open_clip_pytorch_model.bin',
779+
mean=OPENAI_CLIP_MEAN, std=OPENAI_CLIP_STD,
780+
input_size=(3, 320, 320), pool_size=(10, 10), crop_pct=1.0, num_classes=768),
781+
'convnext_large_mlp.clip_laion2b_ft_soup_320': _cfg(
782+
hf_hub_id='laion/CLIP-convnext_large_d_320.laion2B-s29B-b131K-ft-soup',
783+
hf_hub_filename='open_clip_pytorch_model.bin',
784+
mean=OPENAI_CLIP_MEAN, std=OPENAI_CLIP_STD,
785+
input_size=(3, 320, 320), pool_size=(10, 10), crop_pct=1.0, num_classes=768),
776786
})
777787

778788

0 commit comments

Comments
 (0)