mosaicml
/

mosaic-bert-base-seqlen-1024

Model card Files Files and versions Community

Update config.json

#4

by clarine - opened Apr 2

base: refs/heads/main

←

from: refs/pr/4

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