WillHeld commited on
Commit
03f8f02
·
verified ·
1 Parent(s): e0ebee5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -25,6 +25,7 @@ def predict(message, history, temperature, top_p):
25
  "top_p": float(top_p),
26
  "do_sample": True,
27
  "streamer": streamer,
 
28
  }
29
 
30
  # Run generation in a separate thread
 
25
  "top_p": float(top_p),
26
  "do_sample": True,
27
  "streamer": streamer,
28
+ "eos_token_id": 128009,
29
  }
30
 
31
  # Run generation in a separate thread