Merge pull request #5185 from chenhuiyu/feature/add-sailorllm-template
Add SailorLLM template
This commit is contained in:
commit
f59c9bef31
|
@ -782,6 +782,20 @@ _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 an AI assistant named Sailor created by Sea AI Lab. "
|
||||||
|
"Your answer should be friendly, unbiased, faithful, informative and detailed."
|
||||||
|
),
|
||||||
|
stop_words=["<|im_end|>"],
|
||||||
|
replace_eos=True,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
_register_template(
|
_register_template(
|
||||||
name="solar",
|
name="solar",
|
||||||
format_user=StringFormatter(slots=["### User:\n{{content}}\n\n### Assistant:\n"]),
|
format_user=StringFormatter(slots=["### User:\n{{content}}\n\n### Assistant:\n"]),
|
||||||
|
|
Loading…
Reference in New Issue