matejulcar commited on
Commit
1548353
1 Parent(s): fc1fa77

updated to support automodel

Browse files
Files changed (1) hide show
  1. config.json +16 -16
config.json CHANGED
@@ -1,19 +1,19 @@
1
  {
2
- "attention_probs_dropout_prob": 0.1,
3
- "directionality": "bidi",
4
- "hidden_act": "gelu",
5
- "hidden_dropout_prob": 0.1,
6
- "hidden_size": 768,
7
- "initializer_range": 0.02,
8
- "intermediate_size": 3072,
9
- "max_position_embeddings": 512,
10
- "num_attention_heads": 12,
11
- "num_hidden_layers": 12,
12
- "pooler_fc_size": 768,
13
- "pooler_num_attention_heads": 12,
14
- "pooler_num_fc_layers": 3,
15
- "pooler_size_per_head": 128,
16
- "pooler_type": "first_token_transform",
17
- "type_vocab_size": 2,
18
  "vocab_size": 49601
19
  }
1
  {
2
+ "architectures": [
3
+ "BertForMaskedLM"
4
+ ],
5
+ "attention_probs_dropout_prob": 0.1,
6
+ "hidden_act": "gelu",
7
+ "hidden_dropout_prob": 0.1,
8
+ "hidden_size": 768,
9
+ "initializer_range": 0.02,
10
+ "intermediate_size": 3072,
11
+ "layer_norm_eps": 1e-12,
12
+ "max_position_embeddings": 512,
13
+ "model_type": "bert",
14
+ "num_attention_heads": 12,
15
+ "num_hidden_layers": 12,
16
+ "pad_token_id": 0,
17
+ "type_vocab_size": 2,
18
  "vocab_size": 49601
19
  }