utrobinmv commited on
Commit
30b772e
1 Parent(s): 01b97cd

fix generation_config.json

Browse files
Files changed (1) hide show
  1. generation_config.json +1 -0
generation_config.json CHANGED
@@ -2,6 +2,7 @@
2
  "decoder_start_token_id": 0,
3
  "eos_token_id": 1,
4
  "pad_token_id": 0,
 
5
  "repetition_penalty": 5.0,
6
  "num_beams": 5,
7
  "transformers_version": "4.33.0"
 
2
  "decoder_start_token_id": 0,
3
  "eos_token_id": 1,
4
  "pad_token_id": 0,
5
+ "max_new_tokens": 1024,
6
  "repetition_penalty": 5.0,
7
  "num_beams": 5,
8
  "transformers_version": "4.33.0"