Adding the rest of wrappers so finetuning for classification is supported out of the box
Browse filesAdding the rest of the model wrappers (`AutoModelForSequenceClassification`, `AutoModelForMultipleChoice`, `AutoModelForQuestionAnswering` and `AutoModelForTokenClassification`) so finetuning for sequence, token, and other classification tasks is supported out of the box.
- config.json +5 -1
config.json
CHANGED
|
@@ -7,7 +7,11 @@
|
|
| 7 |
"auto_map": {
|
| 8 |
"AutoConfig": "nomic-ai/nomic-bert-2048--configuration_hf_nomic_bert.NomicBertConfig",
|
| 9 |
"AutoModel": "nomic-ai/nomic-bert-2048--modeling_hf_nomic_bert.NomicBertModel",
|
| 10 |
-
"AutoModelForMaskedLM": "nomic-ai/nomic-bert-2048--modeling_hf_nomic_bert.NomicBertForPreTraining"
|
|
|
|
|
|
|
|
|
|
|
|
|
| 11 |
},
|
| 12 |
"bos_token_id": null,
|
| 13 |
"causal": false,
|
|
|
|
| 7 |
"auto_map": {
|
| 8 |
"AutoConfig": "nomic-ai/nomic-bert-2048--configuration_hf_nomic_bert.NomicBertConfig",
|
| 9 |
"AutoModel": "nomic-ai/nomic-bert-2048--modeling_hf_nomic_bert.NomicBertModel",
|
| 10 |
+
"AutoModelForMaskedLM": "nomic-ai/nomic-bert-2048--modeling_hf_nomic_bert.NomicBertForPreTraining"
|
| 11 |
+
"AutoModelForSequenceClassification": "nomic-ai/nomic-bert-2048--modeling_hf_nomic_bert.NomicBertForSequenceClassification",
|
| 12 |
+
"AutoModelForMultipleChoice": "nomic-ai/nomic-bert-2048--modeling_hf_nomic_bert.NomicBertForMultipleChoice",
|
| 13 |
+
"AutoModelForQuestionAnswering": "nomic-ai/nomic-bert-2048--modeling_hf_nomic_bert.NomicBertForQuestionAnswering",
|
| 14 |
+
"AutoModelForTokenClassification": "nomic-ai/nomic-bert-2048--modeling_hf_nomic_bert.NomicBertForTokenClassification"
|
| 15 |
},
|
| 16 |
"bos_token_id": null,
|
| 17 |
"causal": false,
|