support solar 10.7B #1907

Former-commit-id: ecf9b35c612e5514dd25b0d15835d28447a7437e
2024-01-14 00:30:30 +08:00
parent 2edbe87a8c
commit cfe1e24471
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=[