From 52bfcf4883896469a8a2bb7f25d1a064ea5dd354 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: cba5ac9cfc81f11b97831998ea15def5e0b487c2
---
 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
 )