togethercomputer
/

m2-bert-80M-2k-retrieval

Sentence Similarity

text-classification

Model card Files Files and versions Community

Dan Fu commited on Jan 9

Commit

287f84e

•

1 Parent(s): 52b2ffe

Fix

Files changed (1) hide show

config.json +2 -2

config.json CHANGED Viewed

@@ -2,13 +2,13 @@
     "_name_or_path": "togethercomputer/m2-bert-80M-2k-retrieval",
     "alibi_starting_size": 2048,
     "architectures": [
-        "BertForMaskedLM"
     ],
     "attention_probs_dropout_prob": 0.0,
     "bidirectional": true,
     "auto_map": {
         "AutoConfig": "configuration_bert.BertConfig",
-        "AutoModelForTextEncoding": "bert_layers.BertForTextEncoding",
         "AutoTokenizer": "bert-base-uncased"
     },
     "classifier_dropout": null,

     "_name_or_path": "togethercomputer/m2-bert-80M-2k-retrieval",
     "alibi_starting_size": 2048,
     "architectures": [
+        "BertForSequenceClassification"
     ],
     "attention_probs_dropout_prob": 0.0,
     "bidirectional": true,
     "auto_map": {
         "AutoConfig": "configuration_bert.BertConfig",
+        "AutoModelForSequenceClassification": "bert_layers.BertForTextEncoding",
         "AutoTokenizer": "bert-base-uncased"
     },
     "classifier_dropout": null,