423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-10-16 00:28:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix deepseek template

Browse Source 
Former-commit-id: 1fdbcdad9a1cdb20299350efd87a8e5cb8c625a3
This commit is contained in:
hiyouga 2023-11-05 13:08:46 +08:00
parent ee3b33ff03
commit 2a8892b785
1 changed files with 2 additions and 2 deletions
Show Stats 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|>"