mirror of
https://github.com/hiyouga/LLaMA-Factory.git
synced 2025-08-04 12:42:51 +08:00
support deepseek coder #1378
Former-commit-id: 2a8a25819524e84a5e6e907923c47693f8b7a48d
This commit is contained in:
parent
0d488085b8
commit
04107b7af6
@ -621,3 +621,35 @@ register_template(
|
|||||||
system="",
|
system="",
|
||||||
sep=[]
|
sep=[]
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
r"""
|
||||||
|
Supports: https://huggingface.co/deepseek-ai/deepseek-coder-1.3b-instruct
|
||||||
|
https://huggingface.co/deepseek-ai/deepseek-coder-6.7b-instruct
|
||||||
|
https://huggingface.co/deepseek-ai/deepseek-coder-33b-instruct
|
||||||
|
"""
|
||||||
|
register_template(
|
||||||
|
name="deepseek",
|
||||||
|
prefix=[
|
||||||
|
"{{system}}"
|
||||||
|
],
|
||||||
|
prompt=[
|
||||||
|
"### Instruction:\n{{query}}\n### Response:\n"
|
||||||
|
],
|
||||||
|
system=(
|
||||||
|
"You are an AI programming assistant, utilizing the Deepseek Coder model, "
|
||||||
|
"developed by Deepseek Company, and you only answer questions related to computer science. "
|
||||||
|
"For politically sensitive questions, security and privacy issues, "
|
||||||
|
"and other non-computer science questions, you will refuse to answer."
|
||||||
|
),
|
||||||
|
sep=[
|
||||||
|
"\n",
|
||||||
|
{"token": "<|EOT|>"},
|
||||||
|
"\n"
|
||||||
|
],
|
||||||
|
stop_words=[
|
||||||
|
"<|EOT|>"
|
||||||
|
],
|
||||||
|
efficient_eos=True
|
||||||
|
)
|
||||||
|
Loading…
x
Reference in New Issue
Block a user