Vicent Ahuir Esteve commited on
Commit
0e573a3
1 Parent(s): cd6a4ad

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +18 -1
config.json CHANGED
@@ -26,5 +26,22 @@
26
  "torch_dtype": "float32",
27
  "transformers_version": "4.18.0",
28
  "use_cache": true,
29
- "vocab_size": 250112
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
30
  }
26
  "torch_dtype": "float32",
27
  "transformers_version": "4.18.0",
28
  "use_cache": true,
29
+ "vocab_size": 250112,
30
+
31
+ "task_specific_params": {
32
+ "summarization": {
33
+ "early_stopping": true,
34
+ "repetition_penalty": 2.0,
35
+ "max_length": 150,
36
+ "min_length": 25,
37
+ "no_repeat_ngram_size": 3,
38
+ "num_beams": 4
39
+ }
40
+ },
41
+ "early_stopping": true,
42
+ "repetition_penalty": 2.0,
43
+ "max_length": 150,
44
+ "min_length": 25,
45
+ "no_repeat_ngram_size": 3,
46
+ "num_beams": 4
47
  }