From 034b6583480e854adc74a2cff2b634b2233521d8 Mon Sep 17 00:00:00 2001 From: hiyouga Date: Sun, 5 Nov 2023 13:08:46 +0800 Subject: [PATCH] fix deepseek template Former-commit-id: d08f5e8a147f1929567d42b6bed8bc998c2a866d --- src/llmtuner/extras/template.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/llmtuner/extras/template.py b/src/llmtuner/extras/template.py index 6112946f..e8fa9bc6 100644 --- 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|>"