fix chatglm3 template

Former-commit-id: 69bcbc9f6c98e4f4ad97ec0306b33ab21923d311
2023-10-27 21:12:06 +08:00
parent 5919832059
commit 6f09f50d02
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
 )