togethercomputer
/

m2-bert-80M-2k-retrieval

Sentence Similarity

text-classification

Model card Files Files and versions Community

Fix automodel

#3

by osanseviero - opened Feb 5

base: refs/heads/main

←

from: refs/pr/3

Discussion Files changed

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",
-	"AutoModel": "bert_layers.BertModel",
         "AutoModelForSequenceClassification": "bert_layers.BertForTextEncoding"
     },
     "classifier_dropout": null,

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