oliverwang15
commited on
Commit
•
50cb6b5
1
Parent(s):
9dc8536
Supporting Sequence Classification
Browse filesSince `Internlm2_5-7b-chat` have the `InternLM2ForSequenceClassification` class, adding it into the config file to avoid errors when loading the model from `AutoModelForSequenceClassification`.
- 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,
|