zhihan1996 commited on
Commit
282b453
1 Parent(s): 4954c8e

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +2 -0
config.json CHANGED
@@ -23,6 +23,8 @@
23
  },
24
  "auto_map": {
25
  "AutoConfig": "configuration_bert.BertConfig",
 
 
26
  "AutoModelForSequenceClassification": "dnabert_layer.DNABertForSequenceClassification"
27
  },
28
  "layer_norm_eps": 1e-12,
 
23
  },
24
  "auto_map": {
25
  "AutoConfig": "configuration_bert.BertConfig",
26
+ "AutoModel": "dnabert_layer.BertModel",
27
+ "AutoModelForMaskedLM": "dnabert_layer.BertForMaskedLM",
28
  "AutoModelForSequenceClassification": "dnabert_layer.DNABertForSequenceClassification"
29
  },
30
  "layer_norm_eps": 1e-12,