From 4f62e1cb243d996d1764c3c86ca234847ad2c022 Mon Sep 17 00:00:00 2001 From: hoshi-hiyouga Date: Fri, 9 Aug 2024 16:27:42 +0800 Subject: [PATCH] Update template.py --- src/llamafactory/data/template.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/src/llamafactory/data/template.py b/src/llamafactory/data/template.py index 0ec95f33..f4d73701 100644 --- a/src/llamafactory/data/template.py +++ b/src/llamafactory/data/template.py @@ -586,7 +586,8 @@ _register_template( _register_template( name="deepseekcoder", format_user=StringFormatter(slots=["### Instruction:\n{{content}}\n### Response:"]), - format_assistant=StringFormatter(slots=["{{content}}\n<|EOT|>"]), + format_assistant=StringFormatter(slots=["\n{{content}}\n<|EOT|>"]), + format_separator=EmptyFormatter(slots=["\n"]), format_prefix=EmptyFormatter(slots=[{"bos_token"}]), default_system=( "You are an AI programming assistant, utilizing the Deepseek Coder model, " @@ -594,8 +595,6 @@ _register_template( "For politically sensitive questions, security and privacy issues, " "and other non-computer science questions, you will refuse to answer\n" ), - stop_words=["<|EOT|>"], - replace_eos=True, )