vilarin commited on
Commit
61c87fd
·
verified ·
1 Parent(s): 0ad18d2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -81,8 +81,7 @@ def stream_chat(
81
  temperature = temperature,
82
  streamer=streamer,
83
  repetition_penalty=penalty,
84
- pad_token_id = 1,
85
- eos_token_id = 50279,
86
  )
87
 
88
  with torch.no_grad():
 
81
  temperature = temperature,
82
  streamer=streamer,
83
  repetition_penalty=penalty,
84
+ eos_token_id = [2, 73440],
 
85
  )
86
 
87
  with torch.no_grad():