system HF staff commited on
Commit
60e2eea
1 Parent(s): 4542bed

Update config.json

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