From eb26bfc2bade43d848587ac16a54c704d97aecb6 Mon Sep 17 00:00:00 2001
From: hiyouga <hiyouga@buaa.edu.cn>
Date: Fri, 11 Aug 2023 13:51:46 +0800
Subject: [PATCH] fix stop word in baichuan template

Former-commit-id: d59f9389590aab570f68ad4898b035741f9fd1c8
---
 src/llmtuner/extras/template.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/llmtuner/extras/template.py b/src/llmtuner/extras/template.py
index 4bf490aa..64c22285 100644
--- a/src/llmtuner/extras/template.py
+++ b/src/llmtuner/extras/template.py
@@ -438,7 +438,7 @@ register_template(
     ],
     sep=[],
     stop_words=[
-        {"token": "<reserved_102>"} # user token
+        "<reserved_102>" # user token
     ],
     use_history=True
 )