mosaicml
/

mosaic-bert-base-seqlen-1024

Model card Files Files and versions Community

clarine commited on Apr 2

Commit

e2a6999

•

1 Parent(s): b9fba86

Update config.json

Add SequenceClassification to auto_map

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": "bert_layers.BertForSequenceClassification"
   },
   "classifier_dropout": null,
   "gradient_checkpointing": false,