From e408af7fb5fda48b31f4861d12cd44f671983dd3 Mon Sep 17 00:00:00 2001 From: LYMDLUT <70597027+LYMDLUT@users.noreply.github.com> Date: Thu, 18 Jan 2024 10:33:09 +0800 Subject: [PATCH] [Fix]: Update end token id in web_demo.py (#611) --- chat/web_demo.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/chat/web_demo.py b/chat/web_demo.py index c2a18df..e2ce03a 100644 --- a/chat/web_demo.py +++ b/chat/web_demo.py @@ -103,7 +103,7 @@ def main(): model=model, tokenizer=tokenizer, prompt=real_prompt, - additional_eos_token_id=103028, + additional_eos_token_id=92542, **asdict(generation_config), ): # Display robot response in chat message container