From a3d47818b7d33aae1a72971a331f680888957ef0 Mon Sep 17 00:00:00 2001
From: hiyouga <467089858@qq.com>
Date: Wed, 4 Sep 2024 03:06:06 +0800
Subject: [PATCH] fix #5344

Former-commit-id: d41d43a7c37cd10e34c9f399d1a346ffaee641c3
---
 src/llamafactory/data/template.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/llamafactory/data/template.py b/src/llamafactory/data/template.py
index 70f24114..a4d62f58 100644
--- 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"
         )