mosaicml
/

mosaic-bert-base-seqlen-2048

Model card Files Files and versions Community

yangwang825 commited on Jun 12

Commit

cbcdf36

•

1 Parent(s): f136a72

Update config.json

Files changed (1) hide show

config.json +2 -1

config.json CHANGED Viewed

@@ -7,7 +7,8 @@
   "attention_probs_dropout_prob": 0.1,
   "auto_map": {
     "AutoConfig": "configuration_bert.BertConfig",
-    "AutoModelForMaskedLM": "bert_layers.BertForMaskedLM"
   },
   "classifier_dropout": null,
   "gradient_checkpointing": false,

   "attention_probs_dropout_prob": 0.1,
   "auto_map": {
     "AutoConfig": "configuration_bert.BertConfig",
+    "AutoModelForMaskedLM": "bert_layers.BertForMaskedLM",
+    "AutoModelForSequenceClassification": "BertForSequenceClassification"
   },
   "classifier_dropout": null,
   "gradient_checkpointing": false,