From 800567cde72c9b716db767ba6d79898243d630a5 Mon Sep 17 00:00:00 2001
From: hiyouga <467089858@qq.com>
Date: Mon, 9 Sep 2024 22:41:28 +0800
Subject: [PATCH] fix mm plugin

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

diff --git a/src/llamafactory/data/mm_plugin.py b/src/llamafactory/data/mm_plugin.py
index f180c93b..e22e2760 100644
--- a/src/llamafactory/data/mm_plugin.py
+++ b/src/llamafactory/data/mm_plugin.py
@@ -12,6 +12,7 @@ from ..extras.packages import is_pillow_available, is_pyav_available
 
 if is_pillow_available():
     from PIL import Image
+    from PIL.Image import Image as ImageObject
 
 
 if is_pyav_available():
@@ -21,7 +22,6 @@ if is_pyav_available():
 if TYPE_CHECKING:
     import torch
     from av.stream import Stream
-    from PIL.Image import Image as ImageObject
     from transformers import PreTrainedTokenizer, ProcessorMixin
     from transformers.image_processing_utils import BaseImageProcessor