Seba213 commited on
Commit
3c83d5e
·
verified ·
1 Parent(s): 1257aa4

Update generation_config.json

Browse files
Files changed (1) hide show
  1. generation_config.json +16 -10
generation_config.json CHANGED
@@ -1,12 +1,18 @@
1
  {
2
- "decoder_start_token_id": 0,
3
- "early_stopping": true,
4
- "eos_token_id": 1,
5
- "length_penalty": 2.0,
6
- "max_length": 200,
7
- "min_length": 30,
8
- "no_repeat_ngram_size": 3,
9
- "num_beams": 4,
10
- "pad_token_id": 0,
11
- "transformers_version": "4.42.3"
 
 
 
 
 
 
12
  }
 
1
  {
2
+ max_length=128,
3
+ min_length=30,
4
+ # length_penalty=1.0,
5
+ # num_beams=8,
6
+ max_new_tokens=40, # max_target_length,
7
+ do_sample=True,
8
+ top_p=0.9,
9
+ early_stopping=True,
10
+ bos_token_id=model.config.bos_token_id,
11
+ decoder_start_token_id=model.config.decoder_start_token_id,
12
+ no_repeat_ngram_size = 3,
13
+ eos_token_id = model.generation_config.eos_token_id,
14
+ pad_token_id = model.generation_config.pad_token_id,
15
+ forced_bos_token_id = 0,
16
+ forced_eos_token_id = 2,
17
+ top_k=0
18
  }