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

[misc] fix constants (#9008)

Browse Source
This commit is contained in:
Yaowei Zheng
2025-08-23 23:04:30 +08:00
committed by GitHub
parent 0ab0be9df2
commit ec41ef08aa
6 changed files with 40 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/llamafactory/data/template.py
View File

@@ -1855,19 +1855,17 @@ register_template(
)
#copied from seed_coder
# copied from seed_coder
register_template(
name="seed_oss",
format_user=StringFormatter(
slots=[{"bos_token"}, "user\n{{content}}", {"eos_token"}, {"bos_token"}, "assistant\n"]
),
format_system=StringFormatter(slots=[{"bos_token"}, "system\n{{content}}", {"eos_token"}]),
format_function=FunctionFormatter(
slots=[{"bos_token"}, "\n{{content}}", {"eos_token"}], tool_format="seed_oss"
),
format_function=FunctionFormatter(slots=[{"bos_token"}, "\n{{content}}", {"eos_token"}], tool_format="seed_oss"),
format_tools=ToolFormatter(tool_format="seed_oss"),
template_class=ReasoningTemplate,
thought_words=("<seed:think>", "</seed:think>")
thought_words=("<seed:think>", "</seed:think>"),
)