aixsatoshi commited on
Commit
b802e39
1 Parent(s): fad5329

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -63,7 +63,7 @@ def stream_chat(message: str, history: list, temperature: float, max_new_tokens:
63
  max_new_tokens=max_new_tokens,
64
  do_sample=True,
65
  temperature=temperature,
66
- #eos_token_id=[2],
67
  )
68
 
69
  thread = Thread(target=model.generate, kwargs=generate_kwargs)
 
63
  max_new_tokens=max_new_tokens,
64
  do_sample=True,
65
  temperature=temperature,
66
+ eos_token_id=[2, 102401],
67
  )
68
 
69
  thread = Thread(target=model.generate, kwargs=generate_kwargs)