panggi commited on
Commit
b1caeea
1 Parent(s): eb2ec3b

add task_specific_params to config

Browse files
Files changed (1) hide show
  1. config.json +12 -0
config.json CHANGED
@@ -22,5 +22,17 @@
22
  "pad_token_id": 0,
23
  "relative_attention_num_buckets": 32,
24
  "use_cache": true,
 
 
 
 
 
 
 
 
 
 
 
 
25
  "vocab_size": 32128
26
  }
 
22
  "pad_token_id": 0,
23
  "relative_attention_num_buckets": 32,
24
  "use_cache": true,
25
+ "task_specific_params": {
26
+ "summarization": {
27
+ "early_stopping": true,
28
+ "length_penalty": 2.0,
29
+ "repetition_penalty": 2.5,
30
+ "max_length": 100,
31
+ "min_length": 50,
32
+ "no_repeat_ngram_size": 2,
33
+ "num_beams": 2,
34
+ "prefix": "summarize: "
35
+ }
36
+ },
37
  "vocab_size": 32128
38
  }