gsarti commited on
Commit
a83df51
1 Parent(s): 3241a93

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +11 -1
config.json CHANGED
@@ -25,5 +25,15 @@
25
  "torch_dtype": "float32",
26
  "transformers_version": "4.15.0",
27
  "use_cache": true,
28
- "vocab_size": 250112
 
 
 
 
 
 
 
 
 
 
29
  }
 
25
  "torch_dtype": "float32",
26
  "transformers_version": "4.15.0",
27
  "use_cache": true,
28
+ "vocab_size": 250112,
29
+ "task_specific_params": {
30
+ "summarization": {
31
+ "early_stopping": true,
32
+ "length_penalty": 2.0,
33
+ "max_length": 142,
34
+ "min_length": 56,
35
+ "no_repeat_ngram_size": 3,
36
+ "num_beams": 4
37
+ }
38
+ }
39
  }