From 4117f388279ca43eb46def195c21e7051aefd0c7 Mon Sep 17 00:00:00 2001
From: hiyouga <hiyouga@buaa.edu.cn>
Date: Fri, 27 Oct 2023 21:12:06 +0800
Subject: [PATCH] fix chatglm3 template

---
 src/llmtuner/extras/template.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/llmtuner/extras/template.py b/src/llmtuner/extras/template.py
index 26480ba2..ac054837 100644
--- 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
 )