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