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 <[email protected]>
- 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,
|