fix inputs

Former-commit-id: 7d535bb8cdf7e81edda81152e63c8cfe6c9dcc9f
2024-11-23 18:25:45 +00:00
parent cd2485f28d
commit 5003820a6a
14 changed files with 148 additions and 95 deletions
--- a/src/llamafactory/data/template.py
+++ b/src/llamafactory/data/template.py
@@ -785,7 +785,7 @@ _register_template(
    stop_words=["<|eot_id|>"],
    replace_eos=True,
    replace_jinja_template=False,
-    mm_plugin=get_mm_plugin(name="mllama", image_token="<image>"),
+    mm_plugin=get_mm_plugin(name="mllama", image_token="<|image|>"),
 )