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

Adding the rest of wrappers so finetuning for classification is supported out of the box

Browse files

Adding 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.

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,