add orion models

Former-commit-id: 6fc2d5cc03
This commit is contained in:
hiyouga
2024-01-22 21:26:53 +08:00
parent 48cab43cb5
commit 9898712a24
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"]),