fix template

2023-07-18 16:37:23 +08:00 · 2023-07-18 16:37:23 +08:00 · d1ae428c6e
parent cadeac0f44
commit d1ae428c6e
1 changed files with 1 additions and 0 deletions
--- a/src/llmtuner/extras/template.py
+++ b/src/llmtuner/extras/template.py
@ -213,6 +213,7 @@ Supports: https://huggingface.co/HuggingFaceH4/starchat-alpha
          https://huggingface.co/HuggingFaceH4/starchat-beta
 """
 register_template(
+    name="starchat",
    prefix="<|system|>\n",
    prompt="<|user|>\n{query}<|end|>\n<|assistant|>\n",
    sep="<|end|>\n",