Enabled the AutoModelForSequenceClassification in the auto_map
Browse filesAdded the `AutoModelForSequenceClassification` in the `auto_map` to enable Phi3SmallForSequenceClassification to be loaded.
- config.json +1 -0
config.json
CHANGED
@@ -7,6 +7,7 @@
|
|
7 |
"auto_map": {
|
8 |
"AutoConfig": "configuration_phi3_small.Phi3SmallConfig",
|
9 |
"AutoModelForCausalLM": "modeling_phi3_small.Phi3SmallForCausalLM",
|
|
|
10 |
"AutoTokenizer": "tokenization_phi3_small.Phi3SmallTokenizer"
|
11 |
},
|
12 |
"blocksparse_block_size": 64,
|
|
|
7 |
"auto_map": {
|
8 |
"AutoConfig": "configuration_phi3_small.Phi3SmallConfig",
|
9 |
"AutoModelForCausalLM": "modeling_phi3_small.Phi3SmallForCausalLM",
|
10 |
+
"AutoModelForSequenceClassification": "modeling_phi3_small.Phi3SmallForSequenceClassification",
|
11 |
"AutoTokenizer": "tokenization_phi3_small.Phi3SmallTokenizer"
|
12 |
},
|
13 |
"blocksparse_block_size": 64,
|