support solar 10.7B #1907

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=[

src/llmtuner/extras/constants.py

@@ -534,6 +534,20 @@ register_model_group(
 )
 
 
+register_model_group(
+    models={
+        "SOLAR-10.7B": {
+            DownloadSource.DEFAULT: "upstage/SOLAR-10.7B-v1.0"
+        },
+        "SOLAR-10.7B-Chat": {
+            DownloadSource.DEFAULT: "upstage/SOLAR-10.7B-Instruct-v1.0",
+            DownloadSource.MODELSCOPE: "AI-ModelScope/SOLAR-10.7B-Instruct-v1.0"
+        }
+    },
+    template="solar"
+)
+
+
 register_model_group(
     models={
         "Skywork-13B-Base": {