diff --git a/src/llamafactory/data/template.py b/src/llamafactory/data/template.py index 5d4b3011..1f859be6 100644 --- a/src/llamafactory/data/template.py +++ b/src/llamafactory/data/template.py @@ -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( name="solar", format_user=StringFormatter(slots=["### User:\n{{content}}\n\n### Assistant:\n"]),