From 03eba6f041ebf43cd5b339fdb417c1d2f721908a Mon Sep 17 00:00:00 2001 From: hiyouga Date: Tue, 18 Jul 2023 16:37:23 +0800 Subject: [PATCH] fix template Former-commit-id: 729053c9cea6254165ae9c8fd7809479b12f735c --- 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",