From 1b6a66ad9c056482af41c1e7814cacff7b899e5b Mon Sep 17 00:00:00 2001 From: LYMDLUT <70597027+LYMDLUT@users.noreply.github.com> Date: Thu, 18 Jan 2024 04:04:52 +0800 Subject: [PATCH] Update end token id inweb_demo.py --- 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