Merge pull request #2423 from mayflower/main

Support for german sft and dpo

Former-commit-id: 8e282e4e6bee6493b1bd38ba239ca49a6a840a92
This commit is contained in:
hoshi-hiyouga
2024-02-07 15:58:20 +08:00
committed by GitHub
2 changed files with 21 additions and 1 deletions

View File

@@ -515,6 +515,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",