Skip to content

Commit 69b8d7b

Browse files
add Meta CLIP2 presets
1 parent befe0a5 commit 69b8d7b

File tree

1 file changed

+56
-1
lines changed

1 file changed

+56
-1
lines changed
Lines changed: 56 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1,56 @@
1-
backbone_presets = {}
1+
"""MetaCLIP 2 model preset configurations."""
2+
3+
backbone_presets = {
4+
"metaclip_2_vit_huge_patch14_224": {
5+
"metadata": {
6+
"description": (
7+
"986 million parameter, 32-layer for vision and 24-layer for "
8+
"text, patch size of 14, image resolution 224x224. MetaCLIP 2 "
9+
"worldwide huge model (ViT-H-14-quickgelu-worldwide) trained on " # noqa
10+
"29B seen pairs with QuickGELU activation."
11+
),
12+
"params": 1858784002,
13+
"path": "metaclip_2",
14+
},
15+
"kaggle_handle": "kaggle://keras/metaclip2/keras/metaclip_2_vit_huge_patch14_224/1",
16+
},
17+
"metaclip_2_vit_huge_patch14_378": {
18+
"metadata": {
19+
"description": (
20+
"986 million parameter, 32-layer for vision and 24-layer for "
21+
"text, patch size of 14, image resolution 378x378. MetaCLIP 2 "
22+
"worldwide huge model (ViT-H-14-378-worldwide) trained on "
23+
"29B seen pairs."
24+
),
25+
"params": 1859389185,
26+
"path": "metaclip_2",
27+
},
28+
"kaggle_handle": "kaggle://keras/metaclip2/keras/metaclip_2_vit_huge_patch14_378/1",
29+
},
30+
"metaclip_2_vit_giant_patch14_224": {
31+
"metadata": {
32+
"description": (
33+
"1.4 billion parameter, 40-layer for vision and 24-layer for "
34+
"text, patch size of 14, image resolution 224x224. MetaCLIP 2 "
35+
"worldwide giant model (ViT-bigG-14-worldwide) trained on "
36+
"29B seen pairs."
37+
),
38+
"params": 3630409985,
39+
"path": "metaclip_2",
40+
},
41+
"kaggle_handle": "kaggle://keras/metaclip2/keras/metaclip_2_vit_giant_patch14_224/1",
42+
},
43+
"metaclip_2_vit_giant_patch14_378": {
44+
"metadata": {
45+
"description": (
46+
"1.4 billion parameter, 40-layer for vision and 24-layer for "
47+
"text, patch size of 14, image resolution 378x378. MetaCLIP 2 "
48+
"worldwide giant model (ViT-bigG-14-378-worldwide) trained on "
49+
"29B seen pairs."
50+
),
51+
"params": 3631197057,
52+
"path": "metaclip_2",
53+
},
54+
"kaggle_handle": "kaggle://keras/metaclip2/keras/metaclip_2_vit_giant_patch14_378/1",
55+
},
56+
}

0 commit comments

Comments
 (0)