Skip to content

Commit 2cd8da8

Browse files
rankaiyxMasterJH5574
andauthored
[ConvTemplate] Update deepseek.py (#3118)
* Update deepseek.py Update system prompts to replicate the official experience as much as possible. * Fix Pylint --------- Co-authored-by: Ruihang Lai <ruihangl@cs.cmu.edu>
1 parent 892d6e8 commit 2cd8da8

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

python/mlc_llm/conversation_template/deepseek.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,9 @@
4141
Conversation(
4242
name="deepseek_r1_qwen",
4343
system_template=f"<|begin▁of▁sentence|>{MessagePlaceholders.SYSTEM.value}",
44-
system_message="You are a helpful assistant.",
44+
system_message="You are Deepseek-R1, an AI assistant created exclusively by the Chinese "
45+
"Company DeepSeek. You'll provide helpful, harmless, and detailed responses to all "
46+
"user inquiries.",
4547
roles={"user": "<|User|>", "assistant": "<|Assistant|>"},
4648
seps=["", "<|end▁of▁sentence|>"],
4749
role_content_sep="",
@@ -55,7 +57,9 @@
5557
Conversation(
5658
name="deepseek_r1_llama",
5759
system_template=f"<|begin▁of▁sentence|>{MessagePlaceholders.SYSTEM.value}",
58-
system_message="You are a helpful assistant.",
60+
system_message="You are Deepseek-R1, an AI assistant created exclusively by the Chinese "
61+
"Company DeepSeek. You'll provide helpful, harmless, and detailed responses to all"
62+
" user inquiries.",
5963
roles={"user": "<|User|>", "assistant": "<|Assistant|>"},
6064
seps=["", "<|end▁of▁sentence|>"],
6165
role_content_sep="",

0 commit comments

Comments
 (0)