robinzixuan commited on
Commit
f4282cd
1 Parent(s): b77edca

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +6 -1
config.json CHANGED
@@ -3,6 +3,12 @@
3
  "architectures": [
4
  "BertForMaskedLM"
5
  ],
 
 
 
 
 
 
6
  "attention_probs_dropout_prob": 0.1,
7
  "classifier_dropout": null,
8
  "gradient_checkpointing": false,
@@ -25,5 +31,4 @@
25
  "type_vocab_size": 2,
26
  "use_cache": true,
27
  "vocab_size": 30522,
28
- "attn_implementation":"OutEffHop",
29
  }
 
3
  "architectures": [
4
  "BertForMaskedLM"
5
  ],
6
+ "auto_map":
7
+ {"AutoConfig": "configuration_bert.BertConfig",
8
+ "AutoModel": "modeling_bert.BertModel",
9
+ "AutoModelForMaskedLM": "modeling_bert.BertForMaskedLM",
10
+ "AutoModelForSequenceClassification": "modeling_bert.BertForSequenceClassification"
11
+ },
12
  "attention_probs_dropout_prob": 0.1,
13
  "classifier_dropout": null,
14
  "gradient_checkpointing": false,
 
31
  "type_vocab_size": 2,
32
  "use_cache": true,
33
  "vocab_size": 30522,
 
34
  }