aho-tai commited on
Commit
ba96fb5
·
verified ·
1 Parent(s): a23df6a

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +4 -0
config.json CHANGED
@@ -39,6 +39,10 @@
39
  "architectures": [
40
  "PixtralVisionModelBatch"
41
  ],
 
 
 
 
42
  "attention_dropout": 0.0,
43
  "head_dim": 64,
44
  "hidden_act": "silu",
 
39
  "architectures": [
40
  "PixtralVisionModelBatch"
41
  ],
42
+ "auto_map": {
43
+ "AutoConfig": "configuration.PixtralVisionModelBatchConfig",
44
+ "AutoModel": "modeling.PixtralVisionModelBatch"
45
+ },
46
  "attention_dropout": 0.0,
47
  "head_dim": 64,
48
  "hidden_act": "silu",