fix #5344

Former-commit-id: 9d445c0b5be5ccc0e6d1979e76a869ddf92d9534
2025-10-14 23:58:11 +08:00 · 2024-09-04 03:06:06 +08:00 · 2024-09-04 03:06:06 +08:00 · f4dd429cbf
commit f4dd429cbf
parent 7435cde2ef
1 changed files with 1 additions and 1 deletions
--- a/src/llamafactory/data/template.py
+++ b/src/llamafactory/data/template.py
@ -349,7 +349,7 @@ def get_template_and_fix_tokenizer(
    name: Optional[str] = None,
    tool_format: Optional[str] = None,
 ) -> Template:
-    if name == "qwen2_vl":
+    if name in ["llava", "paligemma", "qwen2_vl"]:
        require_version(
            "transformers>=4.45.0.dev0", "To fix: pip install git+https://github.com/huggingface/transformers.git"
        )