DanielGallagherIRE commited on
Commit
85b9889
·
verified ·
1 Parent(s): ce88159

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +2 -2
config.json CHANGED
@@ -2,7 +2,7 @@
2
  "actv_fn": "gelu",
3
  "add_token_type_emb": false,
4
  "architectures": [
5
- "BertForMaskedLM"
6
  ],
7
  "attn_dropout_prob": 0.1,
8
  "attn_out_bias": true,
@@ -28,7 +28,7 @@
28
  "mlp_in_bias": true,
29
  "mlp_out_bias": true,
30
  "mlp_type": "mlp",
31
- "model_type": "bert",
32
  "norm_bias": true,
33
  "norm_eps": 1e-12,
34
  "norm_fn": "layer",
 
2
  "actv_fn": "gelu",
3
  "add_token_type_emb": false,
4
  "architectures": [
5
+ "BertBlocksForMaskedLM"
6
  ],
7
  "attn_dropout_prob": 0.1,
8
  "attn_out_bias": true,
 
28
  "mlp_in_bias": true,
29
  "mlp_out_bias": true,
30
  "mlp_type": "mlp",
31
+ "model_type": "bertblocks",
32
  "norm_bias": true,
33
  "norm_eps": 1e-12,
34
  "norm_fn": "layer",