423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-14 19:06:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix mm plugin

Browse Source
This commit is contained in:
hiyouga
2024-09-09 22:41:28 +08:00
parent c93795ae14
commit 6dc2b00fa2
1 changed files with 1 additions and 1 deletions
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