system HF staff commited on
Commit
ebe714a
1 Parent(s): c6922af

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +11 -2
config.json CHANGED
@@ -13,7 +13,7 @@
13
  "decoder_ffn_dim": 4096,
14
  "decoder_layerdrop": 0.0,
15
  "decoder_layers": 12,
16
- "decoder_start_token_id": null,
17
  "do_sample": false,
18
  "dropout": 0.1,
19
  "early_stopping": true,
@@ -55,7 +55,16 @@
55
  "prefix": " ",
56
  "pruned_heads": {},
57
  "repetition_penalty": 1.0,
58
- "task_specific_params": null,
 
 
 
 
 
 
 
 
 
59
  "temperature": 1.0,
60
  "top_k": 50,
61
  "top_p": 1.0,
 
13
  "decoder_ffn_dim": 4096,
14
  "decoder_layerdrop": 0.0,
15
  "decoder_layers": 12,
16
+ "decoder_start_token_id": 2,
17
  "do_sample": false,
18
  "dropout": 0.1,
19
  "early_stopping": true,
 
55
  "prefix": " ",
56
  "pruned_heads": {},
57
  "repetition_penalty": 1.0,
58
+ "task_specific_params": {
59
+ "summarization": {
60
+ "early_stopping": true,
61
+ "length_penalty": 2.0,
62
+ "max_length": 142,
63
+ "min_length": 56,
64
+ "no_repeat_ngram_size": 3,
65
+ "num_beams": 4
66
+ }
67
+ },
68
  "temperature": 1.0,
69
  "top_k": 50,
70
  "top_p": 1.0,