hsuwill000 commited on
Commit
c48b974
1 Parent(s): c3337cd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -36,7 +36,7 @@ def predict(message, history):
36
  generate_kwargs = dict(
37
  model_inputs,
38
  streamer=streamer,
39
- max_new_tokens=1024,
40
  do_sample=True,
41
  top_p=0.95,
42
  top_k=50,
 
36
  generate_kwargs = dict(
37
  model_inputs,
38
  streamer=streamer,
39
+ max_new_tokens=512,
40
  do_sample=True,
41
  top_p=0.95,
42
  top_k=50,