fix deepseek template

2023-11-05 13:08:46 +08:00 · 2023-11-05 13:08:46 +08:00 · d08f5e8a14
parent 2a8a258195
commit d08f5e8a14
1 changed files with 2 additions and 2 deletions
--- a/src/llmtuner/extras/template.py
+++ b/src/llmtuner/extras/template.py
@ -635,7 +635,7 @@ register_template(
        "{{system}}"
    ],
    prompt=[
-        "### Instruction:\n{{query}}\n### Response:\n"
+        "### Instruction:\n{{query}}\n\n### Response:\n"
    ],
    system=(
        "You are an AI programming assistant, utilizing the Deepseek Coder model, "
@ -646,7 +646,7 @@ register_template(
    sep=[
        "\n",
        {"token": "<|EOT|>"},
-        "\n"
+        "\n\n"
    ],
    stop_words=[
        "<|EOT|>"