zhihan1996
commited on
Commit
•
e42558d
1
Parent(s):
faf3caf
Update config.json
Browse files- 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": "
|
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"
|