fix webUI, fix #171 #177

src/llmtuner/extras/template.py

@@ -203,6 +203,18 @@ register_template(
     name="baichuan",
     prefix="",
     prompt="<reserved_102>{query}<reserved_103>",
-    sep="",
+    sep="</s>",
+    use_history=True
+)
+
+
+r"""
+Supports: https://huggingface.co/HuggingFaceH4/starchat-alpha
+          https://huggingface.co/HuggingFaceH4/starchat-beta
+"""
+register_template(
+    prefix="<|system|>\n",
+    prompt="<|user|>\n{query}<|end|>\n<|assistant|>\n",
+    sep="<|end|>\n",
     use_history=True
 )

src/llmtuner/webui/locales.py

@@ -325,6 +325,14 @@ LOCALES = {
             "label": "最大长度"
         }
     },
+    "max_new_tokens": {
+        "en": {
+            "label": "Maximum new tokens"
+        },
+        "zh": {
+            "label": "最大生成长度"
+        }
+    },
     "top_p": {
         "en": {
             "label": "Top-p"