Update app.py

2024-02-02 11:01:06 +08:00 · 2024-02-02 11:01:06 +08:00 · ba9525ba14
parent 5ca49c93b1
commit ba9525ba14
1 changed files with 1 additions and 1 deletions
--- a/demo/app.py
+++ b/demo/app.py
@ -62,7 +62,7 @@ def generate(
        do_sample=False,
        num_beams=1,
        repetition_penalty=repetition_penalty,
-        eos_token_id=32021
+        eos_token_id=tokenizer.eos_token_id
    )
    t = Thread(target=model.generate, kwargs=generate_kwargs)
    t.start()