Add SailorLLM template

This commit is contained in:
Huiyu Chen 2024-08-15 15:10:14 +08:00
parent 625a0e32c4
commit 2502833a77
1 changed files with 11 additions and 0 deletions

View File

@ -782,6 +782,17 @@ _register_template(
)
_register_template(
name="sailor",
format_user=StringFormatter(slots=["<|im_start|>question\n{{content}}<|im_end|>\n<|im_start|>answer\n"]),
format_system=StringFormatter(slots=["<|im_start|>system\n{{content}}<|im_end|>\n"]),
format_separator=EmptyFormatter(slots=["\n"]),
default_system="You are a helpful assistant.",
stop_words=["<|im_end|>"],
replace_eos=True,
)
_register_template(
name="solar",
format_user=StringFormatter(slots=["### User:\n{{content}}\n\n### Assistant:\n"]),