dingzx97
/

lddbert-mlm

Model card Files Files and versions

dingzx97 commited on Aug 8, 2022

Commit

e07482d

·

1 Parent(s): f57c906

Update config.json

Files changed (1) hide show

config.json +2 -1

config.json CHANGED Viewed

@@ -6,7 +6,8 @@
   "attention_dropout": 0.1,
   "auto_map": {
     "AutoConfig": "configuration_lddbert.LddBertConfig",
-    "AutoModelForMaskedLM": "modeling_lddbert.LddBertForMaskedLM"
   },
   "dim": 768,
   "dropout": 0.1,

   "attention_dropout": 0.1,
   "auto_map": {
     "AutoConfig": "configuration_lddbert.LddBertConfig",
+    "AutoModelForMaskedLM": "modeling_lddbert.LddBertForMaskedLM",
+    "AutoModelForSequenceClassification": "modeling_lddbert.LddBertForSequenceClassification"
   },
   "dim": 768,
   "dropout": 0.1,