fix webUI, fix #171 #177

2025-12-16 11:50:35 +08:00 · 2023-07-18 15:51:48 +08:00
parent 12d8a8633f
commit 6f9360c0bd
2 changed files with 21 additions and 1 deletions
--- a/src/llmtuner/extras/template.py
+++ b/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
 )