From ba9525ba144f0104f53ef3283f7843b3d3de600c Mon Sep 17 00:00:00 2001
From: Daya Guo <40300434+guoday@users.noreply.github.com>
Date: Fri, 2 Feb 2024 11:01:06 +0800
Subject: [PATCH] Update app.py

---
 demo/app.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/demo/app.py b/demo/app.py
index dec892f..5bf817a 100644
--- 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()