mosaicml
/

mosaic-bert-base-seqlen-2048

Model card Files Files and versions Community

Update config.json

#5

by yangwang825 - opened Jun 12

base: refs/heads/main

←

from: refs/pr/5

Discussion Files changed

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,