Lukas Pfahler commited on
Commit
0c2389f
1 Parent(s): 8241db4

include automap in config.json

Browse files
Files changed (1) hide show
  1. config.json +7 -0
config.json CHANGED
@@ -2,6 +2,13 @@
2
  "architectures": [
3
  "Kosmos2_5ForConditionalGeneration"
4
  ],
 
 
 
 
 
 
 
5
  "latent_query_num": 2048,
6
  "model_type": "kosmos-2.5",
7
  "text_config": {
 
2
  "architectures": [
3
  "Kosmos2_5ForConditionalGeneration"
4
  ],
5
+ "auto_map": {
6
+ "AutoConfig": "configuration_kosmos2_5.Kosmos2_5Config",
7
+ "AutoProcessor": "processing_kosmos2_5.Kosmos2_5Processor",
8
+ "AutoImageProcessor": "image_processing_kosmos2_5.Kosmos2_5ImageProcessor",
9
+ "AutoModel": "modeling_kosmos2_5.Kosmos2_5Model",
10
+ "AutoModelForVision2Seq": "modeling_kosmos2_5.Kosmos2_5ForConditionalGeneration"
11
+ },
12
  "latent_query_num": 2048,
13
  "model_type": "kosmos-2.5",
14
  "text_config": {