internlm
/

internlm2_5-7b-chat

Text Generation

Model card Files Files and versions Community

Supporting Sequence Classification

#9

by oliverwang15 - opened Jul 17

base: refs/heads/main

←

from: refs/pr/9

Discussion Files changed

Files changed (1) hide show

config.json +2 -1

config.json CHANGED Viewed

@@ -6,7 +6,8 @@
   "auto_map": {
     "AutoConfig": "configuration_internlm2.InternLM2Config",
     "AutoModelForCausalLM": "modeling_internlm2.InternLM2ForCausalLM",
-    "AutoModel": "modeling_internlm2.InternLM2ForCausalLM"
   },
   "bias": false,
   "bos_token_id": 1,

   "auto_map": {
     "AutoConfig": "configuration_internlm2.InternLM2Config",
     "AutoModelForCausalLM": "modeling_internlm2.InternLM2ForCausalLM",
+    "AutoModel": "modeling_internlm2.InternLM2ForCausalLM",
+    "AutoModelForSequenceClassification": "modeling_internlm2.InternLM2ForSequenceClassification"
   },
   "bias": false,
   "bos_token_id": 1,