zhihan1996 commited on
Commit
a794b19
1 Parent(s): 97755ae

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +2 -1
config.json CHANGED
@@ -25,7 +25,8 @@
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,
31
  "length_penalty": 1.0,
 
25
  "AutoConfig": "configuration_bert.BertConfig",
26
  "AutoModel": "dnabert_layer.BertModel",
27
  "AutoModelForMaskedLM": "dnabert_layer.BertForMaskedLM",
28
+ "AutoModelForSequenceClassification": "dnabert_layer.DNABertForSequenceClassification",
29
+ "AutoModelForPreTraining": "dnabert_layer.BertForPreTraining"
30
  },
31
  "layer_norm_eps": 1e-12,
32
  "length_penalty": 1.0,