kenchan0226 commited on
Commit
db3ef81
·
verified ·
1 Parent(s): 113ecfa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -71,6 +71,7 @@ def generate(
71
  temperature=temperature,
72
  num_beams=1,
73
  repetition_penalty=repetition_penalty,
 
74
  )
75
  t = Thread(target=model.generate, kwargs=generate_kwargs)
76
  t.start()
 
71
  temperature=temperature,
72
  num_beams=1,
73
  repetition_penalty=repetition_penalty,
74
+ eos_token_id=tokenizer.eos_token_id,
75
  )
76
  t = Thread(target=model.generate, kwargs=generate_kwargs)
77
  t.start()