zpn versae commited on
Commit
e5cf08a
·
verified ·
1 Parent(s): f752c1e

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>

Files changed (1) hide show
  1. 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,