oliverwang15 commited on
Commit
50cb6b5
1 Parent(s): 9dc8536

Supporting Sequence Classification

Browse files

Since `Internlm2_5-7b-chat` have the `InternLM2ForSequenceClassification` class, adding it into the config file to avoid errors when loading the model from `AutoModelForSequenceClassification`.

Files changed (1) hide show
  1. config.json +2 -1
config.json CHANGED
@@ -6,7 +6,8 @@
6
  "auto_map": {
7
  "AutoConfig": "configuration_internlm2.InternLM2Config",
8
  "AutoModelForCausalLM": "modeling_internlm2.InternLM2ForCausalLM",
9
- "AutoModel": "modeling_internlm2.InternLM2ForCausalLM"
 
10
  },
11
  "bias": false,
12
  "bos_token_id": 1,
 
6
  "auto_map": {
7
  "AutoConfig": "configuration_internlm2.InternLM2Config",
8
  "AutoModelForCausalLM": "modeling_internlm2.InternLM2ForCausalLM",
9
+ "AutoModel": "modeling_internlm2.InternLM2ForCausalLM",
10
+ "AutoModelForSequenceClassification": "modeling_internlm2.InternLM2ForSequenceClassification"
11
  },
12
  "bias": false,
13
  "bos_token_id": 1,