fix hiyouga#5344

yuwangnexusera · Sep 5, 2024 · d5ebb20 · d5ebb20
1 parent a0eb320
commit d5ebb20
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/llamafactory/data/template.py b/src/llamafactory/data/template.py
@@ -385,7 +385,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"
         )