togethercomputer
/

m2-bert-80M-2k-retrieval

Sentence Similarity

text-classification

Model card Files Files and versions Community

Update config.json

#6

by yangwang825 - opened 27 days ago

base: refs/heads/main

←

from: refs/pr/6

Discussion Files changed

Files changed (1) hide show

config.json +2 -1

config.json CHANGED Viewed

@@ -8,7 +8,8 @@
     "bidirectional": true,
     "auto_map": {
         "AutoConfig": "configuration_bert.BertConfig",
-	"AutoModel": "bert_layers.BertModel",
         "AutoModelForSequenceClassification": "bert_layers.BertForTextEncoding"
     },
     "classifier_dropout": null,

     "bidirectional": true,
     "auto_map": {
         "AutoConfig": "configuration_bert.BertConfig",
+    	"AutoModel": "bert_layers.BertModel",
+        "AutoModelForMaskedLM": "bert_layers.BertForMaskedLM",
         "AutoModelForSequenceClassification": "bert_layers.BertForTextEncoding"
     },
     "classifier_dropout": null,