fix chatglm3 template

2023-10-27 21:12:06 +08:00 · 2023-10-27 21:12:06 +08:00 · 4117f38827
parent 4600c29e93
commit 4117f38827
1 changed files with 4 additions and 0 deletions
--- a/src/llmtuner/extras/template.py
+++ b/src/llmtuner/extras/template.py
@ -576,6 +576,10 @@ register_template(
    ],
    system="",
    sep=[],
+    stop_words=[
+        "<|user|>",
+        "<|observation|>"
+    ],
    efficient_eos=True
 )