kabita-choudhary commited on
Commit
8205e8b
1 Parent(s): 8d9ed84

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +10 -1
config.json CHANGED
@@ -59,7 +59,16 @@
59
  "static_position_embeddings": false,
60
  "student_decoder_layers": null,
61
  "student_encoder_layers": null,
62
- "task_specific_params": {},
 
 
 
 
 
 
 
 
 
63
  "torch_dtype": "float32",
64
  "transformers_version": "4.26.0",
65
  "use_cache": true,
 
59
  "static_position_embeddings": false,
60
  "student_decoder_layers": null,
61
  "student_encoder_layers": null,
62
+ "task_specific_params": {
63
+ "summarization": {
64
+ "early_stopping": true,
65
+ "length_penalty": 2.0,
66
+ "max_length": 142,
67
+ "min_length": 56,
68
+ "no_repeat_ngram_size": 3,
69
+ "num_beams": 4
70
+ }
71
+ },
72
  "torch_dtype": "float32",
73
  "transformers_version": "4.26.0",
74
  "use_cache": true,