Upload model

config.json

@@ -12,7 +12,6 @@
     "bos_token_id": 0,
     "dropout": 0.1,
     "eos_token_id": 2,
-    "gradient_checkpointing": false,
     "hidden_act": "quick_gelu",
     "hidden_size": 32,
     "initializer_factor": 1.0,
@@ -23,8 +22,8 @@
     "model_type": "metaclip_2_text_model",
     "num_attention_heads": 4,
     "num_hidden_layers": 5,
-    "projection_dim": 512,
-    "vocab_size": 99
+    "projection_dim": 32,
+    "vocab_size": 1024
   },
   "transformers_version": "4.57.2",
   "vision_config": {
@@ -32,7 +31,6 @@
     "bos_token_id": 0,
     "dropout": 0.1,
     "eos_token_id": 2,
-    "gradient_checkpointing": false,
     "hidden_act": "quick_gelu",
     "hidden_size": 32,
     "image_size": 224,
@@ -47,7 +45,7 @@
     "num_hidden_layers": 5,
     "pad_token_id": 1,
     "patch_size": 32,
-    "projection_dim": 512,
-    "vocab_size": 99
+    "projection_dim": 32,
+    "vocab_size": 1024
   }
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:6c3d652e8cd59e79977b129b8da4ce5d48d2195f2fbe06d4387f6fa0cb08c7cb
-size 785948
+oid sha256:ae93576ab7c428ee4c7bf1256637ffdd20f16162e4f07fe38e5c1c079fc16161
+size 904380