update baichuan template

2023-07-18 16:43:51 +08:00 · 2023-07-18 16:43:51 +08:00 · bdf91846da
parent d1ae428c6e
commit bdf91846da
1 changed files with 1 additions and 1 deletions
--- a/src/llmtuner/extras/template.py
+++ b/src/llmtuner/extras/template.py
@ -202,7 +202,7 @@ Supports: https://huggingface.co/baichuan-inc/Baichuan-13B-Chat
 register_template(
    name="baichuan",
    prefix="",
-    prompt="<reserved_102>{query}<reserved_103>",
+    prompt=" <reserved_102> {query} <reserved_103> ",
    sep="</s>",
    use_history=True
 )