aixsatoshi commited on
Commit
922f584
1 Parent(s): 2e5208e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -68,7 +68,7 @@ def stream_chat(message: str, history: list, temperature: float, max_new_tokens:
68
  max_new_tokens=max_new_tokens,
69
  do_sample=True,
70
  temperature=temperature,
71
- eos_token_id=[128001, 128009],
72
  )
73
 
74
  thread = Thread(target=model.generate, kwargs=generate_kwargs)
 
68
  max_new_tokens=max_new_tokens,
69
  do_sample=True,
70
  temperature=temperature,
71
+ eos_token_id=[2],
72
  )
73
 
74
  thread = Thread(target=model.generate, kwargs=generate_kwargs)