From 7324984127a978b14982aba08c100cbe8a87145e Mon Sep 17 00:00:00 2001 From: Yimi81 <1548222878@qq.com> Date: Mon, 27 May 2024 13:11:25 +0000 Subject: [PATCH] fix yi template Former-commit-id: dc07413e7d0b138c89eacaef17596e83ef226540 --- src/llamafactory/data/template.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/llamafactory/data/template.py b/src/llamafactory/data/template.py index 8332b0cb..f13b303b 100644 --- a/src/llamafactory/data/template.py +++ b/src/llamafactory/data/template.py @@ -872,6 +872,7 @@ _register_template( _register_template( name="yi", format_user=StringFormatter(slots=["<|im_start|>user\n{{content}}<|im_end|>\n<|im_start|>assistant\n"]), + format_system=StringFormatter(slots=["<|im_start|>system\n{{content}}<|im_end|>\n"]), format_separator=EmptyFormatter(slots=["\n"]), stop_words=["<|im_end|>"], replace_eos=True,