Justin Law commited on
Commit
1fcb3b4
1 Parent(s): 45b57c5

Feature(Generation): Improve Generation Config Params

Browse files
Files changed (1) hide show
  1. generation_config.json +9 -1
generation_config.json CHANGED
@@ -3,5 +3,13 @@
3
  "decoder_start_token_id": 0,
4
  "eos_token_id": 1,
5
  "pad_token_id": 0,
6
- "transformers_version": "4.31.0"
 
 
 
 
 
 
 
 
7
  }
 
3
  "decoder_start_token_id": 0,
4
  "eos_token_id": 1,
5
  "pad_token_id": 0,
6
+ "transformers_version": "4.31.0",
7
+ "max_length": 512,
8
+ "max_new_tokens": 512,
9
+ "early_stopping": true,
10
+ "num_beams": 6,
11
+ "temperature": 0.5,
12
+ "top_k": 50,
13
+ "top_p": 0.90,
14
+ "repetition_penalty": 1
15
  }