utrobinmv commited on
Commit
9dbdf21
1 Parent(s): 5108f13

add generate_config

Browse files
Files changed (1) hide show
  1. generation_config.json +3 -1
generation_config.json CHANGED
@@ -2,5 +2,7 @@
2
  "decoder_start_token_id": 0,
3
  "eos_token_id": 1,
4
  "pad_token_id": 0,
5
- "transformers_version": "4.36.2"
 
 
6
  }
 
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"
8
  }