fix stop word in baichuan template

Former-commit-id: cba5ac9cfc81f11b97831998ea15def5e0b487c2
2023-08-11 13:51:46 +08:00
parent 06df3d6fb6
commit 52bfcf4883
1 changed files with 1 additions and 1 deletions
--- 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
 )