fix #1316

Former-commit-id: 88a753fe80e277007bac2264aee24024e18f2314
2023-10-31 11:32:08 +08:00
parent 93b9f74e9f
commit db06fcfc84
3 changed files with 11 additions and 20 deletions
--- a/src/llmtuner/extras/template.py
+++ b/src/llmtuner/extras/template.py
@@ -249,7 +249,7 @@ register_template(
        "{{system}}"
    ],
    prompt=[
-        "Human: {{query}}\nAssistant: "
+        "Human: {{query}}\nAssistant:"
    ],
    system=(
        "A chat between a curious user and an artificial intelligence assistant. "
@@ -272,7 +272,7 @@ register_template(
        "<<SYS>>\n{{system}}\n<</SYS>>\n\n"
    ],
    prompt=[
-        "[INST] {{query}} [/INST] "
+        "[INST] {{query}} [/INST]"
    ],
    system=(
        "You are a helpful, respectful and honest assistant. "
@@ -298,7 +298,7 @@ register_template(
        "<<SYS>>\n{{system}}\n<</SYS>>\n\n"
    ],
    prompt=[
-        "[INST] {{query}} [/INST] "
+        "[INST] {{query}} [/INST]"
    ],
    system="You are a helpful assistant. 你是一个乐于助人的助手。",
    sep=[]