From 66f719dd96f46e902feac271d8faae69f0730b54 Mon Sep 17 00:00:00 2001
From: Kingsley <82590017+Kuangdd01@users.noreply.github.com>
Date: Fri, 16 May 2025 17:41:04 +0800
Subject: [PATCH] [data] add forward compatibility for video_utils in
 Transformers 4.52.0 (#8077)

---
 src/llamafactory/data/mm_plugin.py | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/src/llamafactory/data/mm_plugin.py b/src/llamafactory/data/mm_plugin.py
index 2837f69f..cd987672 100644
--- a/src/llamafactory/data/mm_plugin.py
+++ b/src/llamafactory/data/mm_plugin.py
@@ -58,7 +58,18 @@ if is_transformers_version_greater_than("4.45.0"):
 
 
 if is_transformers_version_greater_than("4.49.0"):
-    from transformers.image_utils import make_batched_videos, make_flat_list_of_images
+    try:
+        from transformers.image_utils import make_batched_videos, make_flat_list_of_images
+    except ImportError:
+        try:
+            # If that fails, try importing from the new location
+            from transformers.video_utils import make_batched_videos
+            from transformers.image_utils import make_flat_list_of_images
+        except ImportError:
+            raise ImportError(
+                "Could not import make_batched_videos and make_flat_list_of_images. "
+                "In Transformers 4.52.0, make_batched_videos will be moved to transformers.video_utils."
+            )
 
 
 if TYPE_CHECKING: