Update config.json
Browse files- config.json +7 -3
config.json
CHANGED
@@ -4,14 +4,18 @@
|
|
4 |
"MistralForTokenClassification"
|
5 |
],
|
6 |
"attention_dropout": 0.0,
|
|
|
|
|
|
|
7 |
"auto_map": {
|
8 |
-
"AutoConfig":
|
9 |
"AutoModel": "LeroyDyer/Mixtral_AI_PsycoTron--modeling_mistral.MistralModel",
|
10 |
"AutoModelForCausalLM": "LeroyDyer/Mixtral_AI_PsycoTron--modeling_mistral.MistralForCausalLM",
|
|
|
11 |
"AutoModelForQuestionAnswering": "LeroyDyer/Mixtral_AI_PsycoTron--modeling_mistral.MistralForQuestionAnswering",
|
12 |
-
"AutoModelForTokenClassification": "LeroyDyer/Mixtral_AI_PsycoTron--modeling_mistral.MistralForTokenClassification",
|
13 |
"MistralForCausalThoughtsLM": "LeroyDyer/Mixtral_AI_PsycoTron--modeling_mistral.MistralForCausalThoughtsLM",
|
14 |
-
"MistralThoughtsConfig":
|
|
|
15 |
},
|
16 |
"bos_token_id": 1,
|
17 |
"eos_token_id": 2,
|
|
|
4 |
"MistralForTokenClassification"
|
5 |
],
|
6 |
"attention_dropout": 0.0,
|
7 |
+
"architectures": [
|
8 |
+
"MistralForCausalLM","MistralForTokenClassification","MistralForCausalThoughtsLM","MistralForQuestionAnswering","MistralForTranslation"
|
9 |
+
],
|
10 |
"auto_map": {
|
11 |
+
"AutoConfig":"LeroyDyer/Mixtral_AI_PsycoTron--configuration_mistral.MistralConfig",
|
12 |
"AutoModel": "LeroyDyer/Mixtral_AI_PsycoTron--modeling_mistral.MistralModel",
|
13 |
"AutoModelForCausalLM": "LeroyDyer/Mixtral_AI_PsycoTron--modeling_mistral.MistralForCausalLM",
|
14 |
+
"AutoModelForTokenClassification": "LeroyDyer/Mixtral_AI_PsycoTron--modeling_mistral.MistralForTokenClassification",
|
15 |
"AutoModelForQuestionAnswering": "LeroyDyer/Mixtral_AI_PsycoTron--modeling_mistral.MistralForQuestionAnswering",
|
|
|
16 |
"MistralForCausalThoughtsLM": "LeroyDyer/Mixtral_AI_PsycoTron--modeling_mistral.MistralForCausalThoughtsLM",
|
17 |
+
"MistralThoughtsConfig":"LeroyDyer/Mixtral_AI_PsycoTron--configuration_mistral.MistralThoughtsConfig",
|
18 |
+
"AutoModelForTranslation": "LeroyDyer/Mixtral_AI_PsycoTron--modeling_mistral.MistralForTranslation"
|
19 |
},
|
20 |
"bos_token_id": 1,
|
21 |
"eos_token_id": 2,
|