File size: 179 Bytes
4df8249
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
generation_config:
    temperature: 0.95
    top_p: 0.9
    top_k: 50
    num_beams: 1
    use_cache: True
    repetition_penalty: 1.2
    max_new_tokens: 1024
    do_sample: True