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,
|