neuralbioinfo
/

prokbert-mini-c

sequence embedding

genomic language models

promoter-prediction

Model card Files Files and versions Community

ligeti commited on Jan 7

Commit

1f5a70e

·

verified ·

1 Parent(s): 269b7a0

Update config.json

Files changed (1) hide show

config.json +4 -2

config.json CHANGED Viewed

@@ -4,9 +4,11 @@
     "ProkBertForMaskedLM"
   ],
   "attention_probs_dropout_prob": 0.1,
-  "auto_map": {
     "AutoConfig": "models.ProkBertConfig",
-    "AutoModelForMaskedLM": "models.ProkBertForMaskedLM"
   },
   "classification_dropout_rate": 0.1,
   "hidden_act": "gelu",

     "ProkBertForMaskedLM"
   ],
   "attention_probs_dropout_prob": 0.1,
+    "auto_map": {
     "AutoConfig": "models.ProkBertConfig",
+    "AutoModel": "models.ProkBertModel",
+    "AutoModelForMaskedLM": "models.ProkBertForMaskedLM",
+    "AutoModelForSequenceClassification": "models.ProkBertForSequenceClassification"
   },
   "classification_dropout_rate": 0.1,
   "hidden_act": "gelu",