YanshekWoo commited on
Commit
0e59ad2
1 Parent(s): ec80ab7

update generate parameters

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -38,8 +38,7 @@ def chat_func(input_utterance: str, history: Optional[List[str]] = None):
38
 
39
  output_ids = model.generate(input_ids,
40
  max_new_tokens=30,
41
- top_p=0.95,
42
- do_sample=True,
43
  num_beams=4,
44
  repetition_penalty=1.2,
45
  no_repeat_ngram_size=4)[0]
 
38
 
39
  output_ids = model.generate(input_ids,
40
  max_new_tokens=30,
41
+ top_k=32,
 
42
  num_beams=4,
43
  repetition_penalty=1.2,
44
  no_repeat_ngram_size=4)[0]