LeroyDyer
/

Mixtral_AI_TokenClassification

Token Classification

text-generation

Inference Endpoints

text-generation-inference

Model card Files Files and versions Community

LeroyDyer commited on Apr 11

Commit

f94348a

•

1 Parent(s): 12e1e0b

Update config.json

Files changed (1) hide show

config.json +7 -3

config.json CHANGED Viewed

@@ -4,14 +4,18 @@
     "MistralForTokenClassification"
   ],
   "attention_dropout": 0.0,
   "auto_map": {
-    "AutoConfig": "LeroyDyer/Mixtral_AI_PsycoTron--configuration_mistral.MistralConfig",
     "AutoModel": "LeroyDyer/Mixtral_AI_PsycoTron--modeling_mistral.MistralModel",
     "AutoModelForCausalLM": "LeroyDyer/Mixtral_AI_PsycoTron--modeling_mistral.MistralForCausalLM",
     "AutoModelForQuestionAnswering": "LeroyDyer/Mixtral_AI_PsycoTron--modeling_mistral.MistralForQuestionAnswering",
-    "AutoModelForTokenClassification": "LeroyDyer/Mixtral_AI_PsycoTron--modeling_mistral.MistralForTokenClassification",
     "MistralForCausalThoughtsLM": "LeroyDyer/Mixtral_AI_PsycoTron--modeling_mistral.MistralForCausalThoughtsLM",
-    "MistralThoughtsConfig": "LeroyDyer/Mixtral_AI_PsycoTron--configuration_mistral.MistralThoughtsConfig"
   },
   "bos_token_id": 1,
   "eos_token_id": 2,

     "MistralForTokenClassification"
   ],
   "attention_dropout": 0.0,
+  "architectures": [
+    "MistralForCausalLM","MistralForTokenClassification","MistralForCausalThoughtsLM","MistralForQuestionAnswering","MistralForTranslation"
+  ],
   "auto_map": {
+    "AutoConfig":"LeroyDyer/Mixtral_AI_PsycoTron--configuration_mistral.MistralConfig",
     "AutoModel": "LeroyDyer/Mixtral_AI_PsycoTron--modeling_mistral.MistralModel",
     "AutoModelForCausalLM": "LeroyDyer/Mixtral_AI_PsycoTron--modeling_mistral.MistralForCausalLM",
+    "AutoModelForTokenClassification": "LeroyDyer/Mixtral_AI_PsycoTron--modeling_mistral.MistralForTokenClassification",
     "AutoModelForQuestionAnswering": "LeroyDyer/Mixtral_AI_PsycoTron--modeling_mistral.MistralForQuestionAnswering",
     "MistralForCausalThoughtsLM": "LeroyDyer/Mixtral_AI_PsycoTron--modeling_mistral.MistralForCausalThoughtsLM",
+    "MistralThoughtsConfig":"LeroyDyer/Mixtral_AI_PsycoTron--configuration_mistral.MistralThoughtsConfig",
+    "AutoModelForTranslation": "LeroyDyer/Mixtral_AI_PsycoTron--modeling_mistral.MistralForTranslation"
   },
   "bos_token_id": 1,
   "eos_token_id": 2,