423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-14 19:06:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix deepseek template

Browse Source
This commit is contained in:
hiyouga
2023-11-05 13:08:46 +08:00
parent 2a8a258195
commit d08f5e8a14
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/llmtuner/extras/template.py
View File

@@ -635,7 +635,7 @@ register_template(
"{{system}}" "{{system}}"
], ],
prompt=[ prompt=[
"### Instruction:\n{{query}}\n### Response:\n" "### Instruction:\n{{query}}\n\n### Response:\n"
], ],
system=( system=(
"You are an AI programming assistant, utilizing the Deepseek Coder model, " "You are an AI programming assistant, utilizing the Deepseek Coder model, "
@@ -646,7 +646,7 @@ register_template(
sep=[ sep=[
"\n", "\n",
{"token": "<|EOT|>"}, {"token": "<|EOT|>"},
"\n" "\n\n"
], ],
stop_words=[ stop_words=[
"<|EOT|>" "<|EOT|>"