zhihan1996
/

DNABERT-2-117M

Inference Endpoints

Model card Files Files and versions Community

zhihan1996 commited on Feb 22

Commit

1c51ecc

•

1 Parent(s): e42558d

Update config.json

Files changed (1) hide show

config.json +1 -1

config.json CHANGED Viewed

@@ -2,7 +2,7 @@
   "alibi_starting_size": 512,
   "attention_probs_dropout_prob": 0.0,
   "auto_map":
-      {"AutoConfig": "PretrainedConfig",
         "AutoModel": "bert_layers.BertModel",
         "AutoModelForMaskedLM": "bert_layers.BertForMaskedLM",
         "AutoModelForSequenceClassification": "bert_layers.BertForSequenceClassification"

   "alibi_starting_size": 512,
   "attention_probs_dropout_prob": 0.0,
   "auto_map":
+      {"AutoConfig": "configuration_bert.BertConfig",
         "AutoModel": "bert_layers.BertModel",
         "AutoModelForMaskedLM": "bert_layers.BertForMaskedLM",
         "AutoModelForSequenceClassification": "bert_layers.BertForSequenceClassification"