Update config.json
Browse files- config.json +3 -3
config.json
CHANGED
@@ -3,8 +3,8 @@
|
|
3 |
"MistralForCausalLM"
|
4 |
],
|
5 |
"auto_map": {
|
6 |
-
"AutoConfig": "
|
7 |
-
"AutoModel": "
|
8 |
},
|
9 |
"bos_token_id": 1,
|
10 |
"eos_token_id": 2,
|
@@ -13,7 +13,7 @@
|
|
13 |
"initializer_range": 0.02,
|
14 |
"intermediate_size": 14336,
|
15 |
"max_position_embeddings": 32768,
|
16 |
-
"model_type": "
|
17 |
"num_attention_heads": 32,
|
18 |
"num_hidden_layers": 32,
|
19 |
"num_key_value_heads": 8,
|
|
|
3 |
"MistralForCausalLM"
|
4 |
],
|
5 |
"auto_map": {
|
6 |
+
"AutoConfig": "configuration_mistral.mistralConfig",
|
7 |
+
"AutoModel": "modeling_mistral.mistralModel"
|
8 |
},
|
9 |
"bos_token_id": 1,
|
10 |
"eos_token_id": 2,
|
|
|
13 |
"initializer_range": 0.02,
|
14 |
"intermediate_size": 14336,
|
15 |
"max_position_embeddings": 32768,
|
16 |
+
"model_type": "mistral",
|
17 |
"num_attention_heads": 32,
|
18 |
"num_hidden_layers": 32,
|
19 |
"num_key_value_heads": 8,
|