|
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