VictorSanh commited on
Commit
b0eec2b
1 Parent(s): b2fcf7c

add auto_map

Browse files
Files changed (1) hide show
  1. config.json +4 -0
config.json CHANGED
@@ -9,6 +9,10 @@
9
  "Img2HTMLForVisionText2Text"
10
  ],
11
  "attention_dropout": 0.0,
 
 
 
 
12
  "bos_token_id": 1,
13
  "cross_layer_interval": 1,
14
  "eos_token_id": 2,
 
9
  "Img2HTMLForVisionText2Text"
10
  ],
11
  "attention_dropout": 0.0,
12
+ "auto_map": {
13
+ "AutoProcessor": "processing_img2html.Img2HTMLProcessor",
14
+ "AutoModelForCausalLM": "modeling_img2html.Img2HTMLForVisionText2Text"
15
+ },
16
  "bos_token_id": 1,
17
  "cross_layer_interval": 1,
18
  "eos_token_id": 2,