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