jienengchen
/

ViTamin-S

Feature Extraction

Model card Files Files and versions

bbexx commited on Apr 8, 2024

Commit

7b4ed1a

·

1 Parent(s): 393d0cc

update

Files changed (1) hide show

config.json +4 -4

config.json CHANGED Viewed

@@ -7,9 +7,9 @@
       "AutoConfig": "configuration_vitamin.ViTaminConfig",
       "AutoModel": "model.ViTaminCLIP"
     },
-    "embed_dim": 768,
     "vision_cfg": {
-      "timm_model_name": "vitamin_base",
       "timm_model_pretrained": false,
       "timm_pool": "",
       "timm_proj": "linear",
@@ -20,8 +20,8 @@
     "text_cfg": {
       "context_length": 77,
       "vocab_size": 49408,
-      "width": 768,
-      "heads": 12,
       "layers": 12
   }
 }

       "AutoConfig": "configuration_vitamin.ViTaminConfig",
       "AutoModel": "model.ViTaminCLIP"
     },
+    "embed_dim": 384,
     "vision_cfg": {
+      "timm_model_name": "vitamin_small",
       "timm_model_pretrained": false,
       "timm_pool": "",
       "timm_proj": "linear",
     "text_cfg": {
       "context_length": 77,
       "vocab_size": 49408,
+      "width": 384,
+      "heads": 6,
       "layers": 12
   }
 }