[Fix]: Update end token id in web_demo.py (#611)

2024-01-18 10:33:09 +08:00 · 2024-01-18 10:33:09 +08:00 · e408af7fb5
parent 39ffce9090
commit e408af7fb5
1 changed files with 1 additions and 1 deletions
--- 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