MaxLSB commited on
Commit
b56c1a4
·
verified ·
1 Parent(s): 7095d60

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -37,6 +37,7 @@ def respond(
37
  do_sample=True,
38
  temperature=temperature,
39
  top_p=top_p,
 
40
  )
41
 
42
  thread = threading.Thread(target=model.generate, kwargs=generate_kwargs)
 
37
  do_sample=True,
38
  temperature=temperature,
39
  top_p=top_p,
40
+ eos_token_id=tokenizer.eos_token_id,
41
  )
42
 
43
  thread = threading.Thread(target=model.generate, kwargs=generate_kwargs)