add orion models

Former-commit-id: a34db89d2a281d1a1ace29dfd5bd5d4ff7c2f657
This commit is contained in:
hiyouga
2024-01-22 21:26:53 +08:00
parent fe4d93c6db
commit 1fe1ca1c8b
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"]),