Adding the rest of wrappers so finetuning for classification is supported out of the box (#53)
Browse files- Adding the rest of wrappers so finetuning for classification is supported out of the box (a4ea85fdd2e2649328bed284d32b9ea2f1141510)
- Missing one comma (161a1515eaa28482dfe6ceae96cdefdd20dabcbd)
Co-authored-by: Javier de la Rosa <versae@users.noreply.huggingface.co>
- 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,
|