zqh11 commited on
Commit
4062d84
1 Parent(s): 58991d7

Update app.py

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