fix template

Former-commit-id: 01ca056965bb63c5e62ec53ffdb51a062bf64371
2025-08-23 06:12:50 +08:00 · 2024-09-29 22:56:36 +08:00 · 2024-09-29 22:56:36 +08:00 · f1b16236a4
commit f1b16236a4
parent 1c69eea995
1 changed files with 1 additions and 0 deletions
--- 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="<image>"),
 )