FlameF0X commited on
Commit
e348631
·
verified ·
1 Parent(s): 23b35bc

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +6 -2
config.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "architectures": [
3
- "i3Model"
4
  ],
5
  "model_type": "i3",
6
  "vocab_size": 4466,
@@ -13,5 +13,9 @@
13
  "tokenizer_type": "chunk",
14
  "chunk_strategy": "variable_2_3",
15
  "torch_dtype": "float32",
16
- "transformers_version": "4.36.0"
 
 
 
 
17
  }
 
1
  {
2
  "architectures": [
3
+ "I3ForCausalLM"
4
  ],
5
  "model_type": "i3",
6
  "vocab_size": 4466,
 
13
  "tokenizer_type": "chunk",
14
  "chunk_strategy": "variable_2_3",
15
  "torch_dtype": "float32",
16
+ "transformers_version": "4.36.0",
17
+ "auto_map": {
18
+ "AutoConfig": "configuration_i3.I3Config",
19
+ "AutoModelForCausalLM": "modeling_i3.I3ForCausalLM"
20
+ }
21
  }