From 5ed2a97056da4258f3c6f121b790feb1545edaf0 Mon Sep 17 00:00:00 2001 From: hiyouga Date: Tue, 18 Jul 2023 16:43:51 +0800 Subject: [PATCH] update baichuan template Former-commit-id: 03520588c39986c98a0515a64993af8c2468b9d0 --- src/llmtuner/extras/template.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/llmtuner/extras/template.py b/src/llmtuner/extras/template.py index 665a684c..26e62101 100644 --- a/src/llmtuner/extras/template.py +++ b/src/llmtuner/extras/template.py @@ -202,7 +202,7 @@ Supports: https://huggingface.co/baichuan-inc/Baichuan-13B-Chat register_template( name="baichuan", prefix="", - prompt="{query}", + prompt=" {query} ", sep="", use_history=True )