YanshekWoo commited on
Commit
ec80ab7
1 Parent(s): de1ff8e

update generate parameters

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -41,7 +41,8 @@ def chat_func(input_utterance: str, history: Optional[List[str]] = None):
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
 
 
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]
46
 
47
  response = tokenizer.decode(output_ids, skip_special_tokens=True)
48