Qian Liu commited on
Commit
09df58c
1 Parent(s): df3358a

Upload config.json

Browse files

Keep aligned with bart-base on forced_bos_token_id

Files changed (1) hide show
  1. config.json +1 -21
config.json CHANGED
@@ -23,6 +23,7 @@
23
  "encoder_layerdrop": 0.0,
24
  "encoder_layers": 6,
25
  "eos_token_id": 2,
 
26
  "forced_eos_token_id": 2,
27
  "gradient_checkpointing": false,
28
  "id2label": {
@@ -39,33 +40,12 @@
39
  },
40
  "max_position_embeddings": 1024,
41
  "model_type": "bart",
42
- "no_repeat_ngram_size": 3,
43
  "normalize_before": false,
44
  "normalize_embedding": true,
45
  "num_beams": 4,
46
  "num_hidden_layers": 6,
47
  "pad_token_id": 1,
48
  "scale_embedding": false,
49
- "task_specific_params": {
50
- "summarization": {
51
- "length_penalty": 1.0,
52
- "max_length": 128,
53
- "min_length": 12,
54
- "num_beams": 4
55
- },
56
- "summarization_cnn": {
57
- "length_penalty": 2.0,
58
- "max_length": 142,
59
- "min_length": 56,
60
- "num_beams": 4
61
- },
62
- "summarization_xsum": {
63
- "length_penalty": 1.0,
64
- "max_length": 62,
65
- "min_length": 11,
66
- "num_beams": 6
67
- }
68
- },
69
  "torch_dtype": "float32",
70
  "transformers_version": "4.17.0.dev0",
71
  "use_cache": true,
 
23
  "encoder_layerdrop": 0.0,
24
  "encoder_layers": 6,
25
  "eos_token_id": 2,
26
+ "forced_bos_token_id": 0,
27
  "forced_eos_token_id": 2,
28
  "gradient_checkpointing": false,
29
  "id2label": {
 
40
  },
41
  "max_position_embeddings": 1024,
42
  "model_type": "bart",
 
43
  "normalize_before": false,
44
  "normalize_embedding": true,
45
  "num_beams": 4,
46
  "num_hidden_layers": 6,
47
  "pad_token_id": 1,
48
  "scale_embedding": false,
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
49
  "torch_dtype": "float32",
50
  "transformers_version": "4.17.0.dev0",
51
  "use_cache": true,