togethercomputer
/

m2-bert-80M-8k-retrieval

Sentence Similarity

text-classification

Model card Files Files and versions

danfu09 commited on Jan 13, 2024

Commit

90e0b28

·

verified ·

1 Parent(s): 7d69cb9

Update config.json

Files changed (1) hide show

config.json +1 -1

config.json CHANGED Viewed

@@ -9,7 +9,7 @@
     "auto_map": {
         "AutoConfig": "configuration_bert.BertConfig",
         "AutoModel": "bert_layers.BertModel",
-        "AutoModelForSequenceClassification": "bert_layers.BertForTextEncoding",
     },
     "classifier_dropout": null,
     "gradient_checkpointing": false,

     "auto_map": {
         "AutoConfig": "configuration_bert.BertConfig",
         "AutoModel": "bert_layers.BertModel",
+        "AutoModelForSequenceClassification": "bert_layers.BertForTextEncoding"
     },
     "classifier_dropout": null,
     "gradient_checkpointing": false,