system HF staff commited on
Commit
61de1ba
1 Parent(s): 6815fb0

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +4 -4
config.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "architectures": [
3
- "RobertaModel"
4
  ],
5
  "attention_probs_dropout_prob": 0.1,
6
  "bos_token_id": 0,
@@ -11,12 +11,12 @@
11
  "hidden_size": 768,
12
  "initializer_range": 0.02,
13
  "intermediate_size": 3072,
14
- "layer_norm_eps": 1e-05,
15
  "max_position_embeddings": 514,
16
  "model_type": "roberta",
17
  "num_attention_heads": 12,
18
- "num_hidden_layers": 12,
19
  "pad_token_id": 1,
20
  "type_vocab_size": 1,
21
- "vocab_size": 50265
22
  }
1
  {
2
  "architectures": [
3
+ "RobertaForMaskedLM"
4
  ],
5
  "attention_probs_dropout_prob": 0.1,
6
  "bos_token_id": 0,
11
  "hidden_size": 768,
12
  "initializer_range": 0.02,
13
  "intermediate_size": 3072,
14
+ "layer_norm_eps": 1e-12,
15
  "max_position_embeddings": 514,
16
  "model_type": "roberta",
17
  "num_attention_heads": 12,
18
+ "num_hidden_layers": 6,
19
  "pad_token_id": 1,
20
  "type_vocab_size": 1,
21
+ "vocab_size": 52000
22
  }