yaya-sy commited on
Commit
8845f00
Β·
verified Β·
1 Parent(s): 00d1062

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -102,7 +102,7 @@ def model_inference(input_dict, history):
102
  ).to("cuda")
103
  # Set up streaming generation.
104
  streamer = TextIteratorStreamer(processor, skip_prompt=True, skip_special_tokens=True)
105
- generation_kwargs = dict(inputs, streamer=streamer, max_new_tokens=256, min_p=0.01, no_repeat_ngram_size=1, temperature=4, num_beams=5, repetition_penalty=1.5)
106
  thread = Thread(target=model.generate, kwargs=generation_kwargs)
107
  thread.start()
108
  buffer = ""
 
102
  ).to("cuda")
103
  # Set up streaming generation.
104
  streamer = TextIteratorStreamer(processor, skip_prompt=True, skip_special_tokens=True)
105
+ generation_kwargs = dict(inputs, streamer=streamer, max_new_tokens=256, no_repeat_ngram_size=20, temperature=2, num_beams=5, repetition_penalty=1.5)
106
  thread = Thread(target=model.generate, kwargs=generation_kwargs)
107
  thread.start()
108
  buffer = ""