aho-tai
/

PixtralEncoderDecoder-v0

vision_pixtral_encoder_decoder

Model card Files Files and versions

aho-tai commited on May 6

Commit

b3986fe

·

verified ·

1 Parent(s): 27ca9d0

Update config.json

Files changed (1) hide show

config.json +6 -2

config.json CHANGED Viewed

@@ -29,10 +29,14 @@
   },
   "encoder": {
     "_attn_implementation_autoset": true,
-    "_name_or_path": "pixtral-vision",
     "architectures": [
-      "PixtralVisionModel"
     ],
     "attention_dropout": 0.0,
     "head_dim": 64,
     "hidden_act": "silu",

   },
   "encoder": {
     "_attn_implementation_autoset": true,
+    "_name_or_path": "pixtral_encoder_decoder",
     "architectures": [
+      "PixtralVisionModelBatch"
     ],
+    "auto_map": {
+      "AutoConfig": "configuration.PixtralVisionModelBatchConfig",
+      "AutoModel": "modeling.PixtralVisionModelBatch"
+    },
     "attention_dropout": 0.0,
     "head_dim": 64,
     "hidden_act": "silu",