[data] gemma3 plugin pan and scan (#7294)

* gemma3 pan and scan * add test case * fix test
2025-03-13 23:29:23 +08:00
parent 0be0d7796a
commit 93e6184cbe
5 changed files with 65 additions and 4 deletions
--- a/src/llamafactory/data/template.py
+++ b/src/llamafactory/data/template.py
@@ -1263,6 +1263,7 @@ register_template(
    format_user=StringFormatter(slots=["{{content}}\n"]),
    format_prefix=EmptyFormatter(slots=[{"bos_token"}]),
    mm_plugin=get_mm_plugin(name="paligemma", image_token="<image>"),
+    template_class=Llama2Template,
 )


@@ -1277,6 +1278,7 @@ register_template(
    format_prefix=EmptyFormatter(slots=[{"bos_token"}]),
    stop_words=["<end_of_turn>"],
    mm_plugin=get_mm_plugin(name="paligemma", image_token="<image>"),
+    template_class=Llama2Template,
 )