add orion models

This commit is contained in:
hiyouga
2024-01-22 21:26:53 +08:00
parent 486cc8d360
commit 6fc2d5cc03
2 changed files with 36 additions and 1 deletions

View File

@@ -462,6 +462,14 @@ register_template(
)
register_template(
name="orion",
format_user=StringFormatter(slots=["Human: {{content}}\n\nAssistant: </s>"]),
format_system=StringFormatter(slots=[{"bos_token"}, "{{content}}"]),
force_system=True,
)
register_template(
name="qwen",
format_user=StringFormatter(slots=["<|im_start|>user\n{{content}}<|im_end|>\n<|im_start|>assistant\n"]),