pumad commited on
Commit
d1d2356
·
verified ·
1 Parent(s): 8f83066

Fix generation_config.json - add num_beams=4

Browse files
Files changed (1) hide show
  1. generation_config.json +13 -5
generation_config.json CHANGED
@@ -1,8 +1,16 @@
1
  {
2
  "_from_model_config": true,
3
- "decoder_start_token_id": 0,
4
- "eos_token_id": 3,
5
- "forced_eos_token_id": 3,
6
- "pad_token_id": 0,
 
 
 
 
 
 
 
 
7
  "transformers_version": "4.57.3"
8
- }
 
1
  {
2
  "_from_model_config": true,
3
+ "bad_words_ids": [
4
+ [
5
+ 65000
6
+ ]
7
+ ],
8
+ "bos_token_id": 65000,
9
+ "decoder_start_token_id": 65000,
10
+ "eos_token_id": 0,
11
+ "forced_eos_token_id": 0,
12
+ "max_length": 512,
13
+ "num_beams": 4,
14
+ "pad_token_id": 65000,
15
  "transformers_version": "4.57.3"
16
+ }