support solar 10.7B #1907

Former-commit-id: bf73224f33
2025-12-16 03:40:34 +08:00 · 2024-01-14 00:30:30 +08:00
parent a994279a3a
commit 21020e51ca
2 changed files with 27 additions and 0 deletions
--- a/src/llmtuner/data/template.py
+++ b/src/llmtuner/data/template.py
@@ -617,6 +617,19 @@ register_template(
 )


+register_template(
+    name="solar",
+    prefix=[
+        "{{system}}"
+    ],
+    prompt=[
+        "### User:\n{{query}}\n\n### Assistant:\n"
+    ],
+    system="",
+    sep=[]
+)
+
+
 register_template(
    name="starchat",
    prefix=[