From d1ae428c6ec2a50e0cfc70657111fc5b67eff51e Mon Sep 17 00:00:00 2001 From: hiyouga Date: Tue, 18 Jul 2023 16:37:23 +0800 Subject: [PATCH] fix template --- src/llmtuner/extras/template.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/llmtuner/extras/template.py b/src/llmtuner/extras/template.py index 52f3f8a1..665a684c 100644 --- 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",