YanshekWoo commited on
Commit
c57292d
1 Parent(s): e79cdfd

update generate parameters

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