gmonsoon commited on
Commit
9fe417b
·
verified ·
1 Parent(s): e7ba0e1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -57,7 +57,7 @@ def generate(
57
  temperature=temperature,
58
  num_beams=1,
59
  repetition_penalty=repetition_penalty,
60
- eos_token_id=tokenizer.eos_token_id,
61
  )
62
  t = Thread(target=model.generate, kwargs=generate_kwargs)
63
  t.start()
 
57
  temperature=temperature,
58
  num_beams=1,
59
  repetition_penalty=repetition_penalty,
60
+ eos_token_id=151643,
61
  )
62
  t = Thread(target=model.generate, kwargs=generate_kwargs)
63
  t.start()