martinkorelic commited on
Commit
ffd6feb
1 Parent(s): fc4d1ba

Update generation_config.json

Browse files
Files changed (1) hide show
  1. generation_config.json +4 -2
generation_config.json CHANGED
@@ -1,10 +1,12 @@
1
  {
2
-
3
  "decoder_start_token_id": 0,
4
  "eos_token_id": 1,
5
  "pad_token_id": 0,
6
  "num_beams": 5,
7
  "early_stopping": true,
 
 
8
  "max_length": 64,
9
  "transformers_version": "4.40.0"
10
- }
 
1
  {
2
+ "_from_model_config": true,
3
  "decoder_start_token_id": 0,
4
  "eos_token_id": 1,
5
  "pad_token_id": 0,
6
  "num_beams": 5,
7
  "early_stopping": true,
8
+ "num_return_sequences": 1,
9
+ "no_repeat_ngram_size": 2,
10
  "max_length": 64,
11
  "transformers_version": "4.40.0"
12
+ }