DeepSeek/Janus
1
0
Fork 0
mirror of https://github.com/deepseek-ai/Janus synced 2025-06-26 18:16:30 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

add Janus-Pro

Browse Source
This commit is contained in:
charlescxk
2025-01-27 23:46:51 +08:00
parent 146668eafe
commit 95bde087ba
5 changed files with 475 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
janus/utils/conversation.py
View File

@@ -275,7 +275,7 @@ register_conv_template(
# deepseek template
register_conv_template(
Conversation(
name="deepseek",
name="deepseek_old",
system_template="{system_message}",
# system_message="You are a helpful assistant. Please answer truthfully and write out your "
# "thinking step by step to be sure you get the right answer.",
@@ -290,6 +290,23 @@ register_conv_template(
stop_str=["User:", "<｜end▁of▁sentence｜>"],
)
)
register_conv_template(
Conversation(
name="deepseek",
system_template="{system_message}",
# system_message="You are a helpful assistant. Please answer truthfully and write out your "
# "thinking step by step to be sure you get the right answer.",
system_message="",
roles=("<|User|>", "<|Assistant|>"),
messages=(),
offset=0,
sep_style=SeparatorStyle.DeepSeek,
sep="\n\n",
sep2="<｜end▁of▁sentence｜>",
stop_token_ids=[100001],
stop_str=["<|User|>", "<｜end▁of▁sentence｜>"]
)
)
register_conv_template(
Conversation(