nomic-ai
/

nomic-bert-2048

Model card Files Files and versions Community

zpn commited on Feb 23

Commit

47fb1ce

•

1 Parent(s): cbfed57

Update config.json

Files changed (1) hide show

config.json +2 -1

config.json CHANGED Viewed

@@ -6,7 +6,8 @@
   "attn_pdrop": 0.0,
   "auto_map": {
     "AutoConfig": "configuration_hf_nomic_bert.NomicBertConfig",
-    "AutoModelForMaskedLM": "modeling_hf_nomic_bert.NomicBertForPreTraining"
   },
   "bos_token_id": null,
   "causal": false,

   "attn_pdrop": 0.0,
   "auto_map": {
     "AutoConfig": "configuration_hf_nomic_bert.NomicBertConfig",
+    "AutoModelForMaskedLM": "modeling_hf_nomic_bert.NomicBertForPreTraining",
+    "AutoModelForSequenceClassification": "modeling_hf_nomic_bert.NomicBertForSequenceClassification"
   },
   "bos_token_id": null,
   "causal": false,