diff --git a/src/llmtuner/data/template.py b/src/llmtuner/data/template.py index a3b23be9..89c704a9 100644 --- a/src/llmtuner/data/template.py +++ b/src/llmtuner/data/template.py @@ -504,6 +504,16 @@ register_template( register_template(name="vanilla") +register_template( + name="chatml_de", + format_user=StringFormatter(slots=["<|im_start|>user\n{{content}}<|im_end|>\n<|im_start|>assistant\n"]), + format_system=StringFormatter(slots=["<|im_start|>system\n{{content}}<|im_end|>\n"]), + format_separator=EmptyFormatter(slots=["\n"]), + default_system="Du bist ein freundlicher und hilfsbereiter KI-Assistent.", + stop_words=["<|im_end|>"], + replace_eos=True, +) + register_template( name="vicuna",