diff --git a/swift/llm/template/template/utils.py b/swift/llm/template/template/utils.py index fab39510ff..bb1072901f 100644 --- a/swift/llm/template/template/utils.py +++ b/swift/llm/template/template/utils.py @@ -15,7 +15,7 @@ class ChatmlTemplateMeta(TemplateMeta): prefix: Prompt = field(default_factory=list) prompt: Prompt = field(default_factory=lambda: ['<|im_start|>user\n{{QUERY}}<|im_end|>\n<|im_start|>assistant\n']) chat_sep: Optional[Prompt] = field(default_factory=lambda: ['<|im_end|>\n']) - suffix: Prompt = field(default_factory=lambda: ['<|im_end|>']) + suffix: Prompt = field(default_factory=lambda: ['<|im_end|>\n']) system_prefix: Optional[Prompt] = field(default_factory=lambda: ['<|im_start|>system\n{{SYSTEM}}<|im_end|>\n']) auto_add_bos: bool = True