From f1b16236a4ba54625a5fa54b3f5145ca5cf3b552 Mon Sep 17 00:00:00 2001 From: BUAADreamer <1428195643@qq.com> Date: Sun, 29 Sep 2024 22:56:36 +0800 Subject: [PATCH] fix template Former-commit-id: 01ca056965bb63c5e62ec53ffdb51a062bf64371 --- 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 3a7bfd5d..7a10a0e3 100644 --- a/src/llamafactory/data/template.py +++ b/src/llamafactory/data/template.py @@ -816,6 +816,7 @@ _register_template( stop_words=["<|im_end|>"], replace_eos=True, replace_jinja_template=False, + mm_plugin=get_mm_plugin(name="llava_next", image_token=""), )