fix: fix the deepseekcoder template to avoid repeat problem

This commit is contained in:
moontidef 2024-08-05 23:55:45 +08:00
parent 1c05b847b2
commit b82ecbedd0
1 changed files with 3 additions and 2 deletions

View File

@ -586,8 +586,7 @@ _register_template(
_register_template(
name="deepseekcoder",
format_user=StringFormatter(slots=["### Instruction:\n{{content}}\n### Response:"]),
format_assistant=StringFormatter(slots=["\n{{content}}\n"]),
format_separator=EmptyFormatter(slots=["\n"]),
format_assistant=StringFormatter(slots=["{{content}}\n<|EOT|>"]),
format_prefix=EmptyFormatter(slots=[{"bos_token"}]),
default_system=(
"You are an AI programming assistant, utilizing the Deepseek Coder model, "
@ -595,6 +594,8 @@ _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,
)