system HF staff commited on
Commit
31046d3
1 Parent(s): d273ac2

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +4 -26
config.json CHANGED
@@ -3,42 +3,20 @@
3
  "RobertaForMaskedLM"
4
  ],
5
  "attention_probs_dropout_prob": 0.1,
6
- "do_sample": false,
7
- "finetuning_task": null,
8
  "hidden_act": "gelu",
9
  "hidden_dropout_prob": 0.1,
10
  "hidden_size": 768,
11
- "id2label": {
12
- "0": "LABEL_0",
13
- "1": "LABEL_1"
14
- },
15
  "initializer_range": 0.02,
16
  "intermediate_size": 3072,
17
- "is_decoder": false,
18
- "label2id": {
19
- "LABEL_0": 0,
20
- "LABEL_1": 1
21
- },
22
  "layer_norm_eps": 1e-05,
23
- "length_penalty": 1.0,
24
- "max_length": 20,
25
  "max_position_embeddings": 514,
26
  "model_type": "roberta",
27
  "num_attention_heads": 12,
28
- "num_beams": 1,
29
  "num_hidden_layers": 6,
30
- "num_labels": 2,
31
- "num_return_sequences": 1,
32
- "output_attentions": false,
33
- "output_hidden_states": false,
34
  "output_past": true,
35
- "pruned_heads": {},
36
- "repetition_penalty": 1.0,
37
- "temperature": 1.0,
38
- "top_k": 50,
39
- "top_p": 1.0,
40
- "torchscript": false,
41
  "type_vocab_size": 1,
42
- "use_bfloat16": false,
43
  "vocab_size": 52000
44
- }
 
3
  "RobertaForMaskedLM"
4
  ],
5
  "attention_probs_dropout_prob": 0.1,
6
+ "bos_token_id": 0,
7
+ "eos_token_id": 2,
8
  "hidden_act": "gelu",
9
  "hidden_dropout_prob": 0.1,
10
  "hidden_size": 768,
 
 
 
 
11
  "initializer_range": 0.02,
12
  "intermediate_size": 3072,
 
 
 
 
 
13
  "layer_norm_eps": 1e-05,
 
 
14
  "max_position_embeddings": 514,
15
  "model_type": "roberta",
16
  "num_attention_heads": 12,
 
17
  "num_hidden_layers": 6,
 
 
 
 
18
  "output_past": true,
19
+ "pad_token_id": 1,
 
 
 
 
 
20
  "type_vocab_size": 1,
 
21
  "vocab_size": 52000
22
+ }