423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-10-14 23:58:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix mm plugin

Browse Source 
Former-commit-id: 6a3549c6c1a8c40de61e748f0b280bfc9e1279a2
This commit is contained in:
hiyouga 2024-09-09 22:41:28 +08:00
parent 7a3ba5a25d
commit 800567cde7
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/llamafactory/data/mm_plugin.py
View File

@ -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