zhihan1996 commited on
Commit
e42558d
1 Parent(s): faf3caf

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -2,7 +2,7 @@
2
  "alibi_starting_size": 512,
3
  "attention_probs_dropout_prob": 0.0,
4
  "auto_map":
5
- {"AutoConfig": "configuration_bert.BertConfig",
6
  "AutoModel": "bert_layers.BertModel",
7
  "AutoModelForMaskedLM": "bert_layers.BertForMaskedLM",
8
  "AutoModelForSequenceClassification": "bert_layers.BertForSequenceClassification"
 
2
  "alibi_starting_size": 512,
3
  "attention_probs_dropout_prob": 0.0,
4
  "auto_map":
5
+ {"AutoConfig": "PretrainedConfig",
6
  "AutoModel": "bert_layers.BertModel",
7
  "AutoModelForMaskedLM": "bert_layers.BertForMaskedLM",
8
  "AutoModelForSequenceClassification": "bert_layers.BertForSequenceClassification"