gsarti commited on
Commit
84c1028
1 Parent(s): 6da80fb

Update config.json

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