YanshekWoo commited on
Commit
de1ff8e
1 Parent(s): 287e49e

update generate parameters

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