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 )