Enabled the AutoModelForSequenceClassification in the auto_map (#22)
Browse files- Enabled the AutoModelForSequenceClassification in the auto_map (32e806687563ec1bc38f25f5605b72ed83dfaed5)
Co-authored-by: Mohammadreza Mohseni <[email protected]>
- 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,
|