add orion models

Former-commit-id: 6fc2d5cc03
2025-12-15 03:10:35 +08:00 · 2024-01-22 21:26:53 +08:00
parent 48cab43cb5
commit 9898712a24
2 changed files with 36 additions and 1 deletions
--- a/src/llmtuner/data/template.py
+++ b/src/llmtuner/data/template.py
@@ -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"]),