togethercomputer
/

m2-bert-80M-2k-retrieval

Sentence Similarity

text-classification

Model card Files Files and versions Community

danfu09 commited on Jan 9

Commit

52b2ffe

•

1 Parent(s): 0f1a7ee

Update config.json

Files changed (1) hide show

config.json +1 -1

config.json CHANGED Viewed

@@ -8,7 +8,7 @@
     "bidirectional": true,
     "auto_map": {
         "AutoConfig": "configuration_bert.BertConfig",
-        "AutoModelForMaskedLM": "bert_layers.BertForMaskedLM",
         "AutoTokenizer": "bert-base-uncased"
     },
     "classifier_dropout": null,

     "bidirectional": true,
     "auto_map": {
         "AutoConfig": "configuration_bert.BertConfig",
+        "AutoModelForTextEncoding": "bert_layers.BertForTextEncoding",
         "AutoTokenizer": "bert-base-uncased"
     },
     "classifier_dropout": null,