umuthopeyildirim commited on
Commit
be76452
·
verified ·
1 Parent(s): 9051e7c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -61,7 +61,7 @@ def predict(message, history):
61
  generate_kwargs = dict(
62
  model_inputs,
63
  streamer=streamer,
64
- max_new_tokens=9999,
65
  do_sample=True,
66
  top_p=0.95,
67
  top_k=1000,
 
61
  generate_kwargs = dict(
62
  model_inputs,
63
  streamer=streamer,
64
+ max_new_tokens=1024,
65
  do_sample=True,
66
  top_p=0.95,
67
  top_k=1000,