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

fix mm plugin

Browse Source 
Former-commit-id: 6dc2b00fa2bcbdbad71cb0d96e14d75a7db932e2
This commit is contained in:
hiyouga 2024-09-09 22:41:28 +08:00
parent c52eeb70e7
commit 2bb1ee3292
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(): if is_pillow_available():
from PIL import Image from PIL import Image
from PIL.Image import Image as ImageObject
if is_pyav_available(): if is_pyav_available():
@ -21,7 +22,6 @@ if is_pyav_available():
if TYPE_CHECKING: if TYPE_CHECKING:
import torch import torch
from av.stream import Stream from av.stream import Stream
from PIL.Image import Image as ImageObject
from transformers import PreTrainedTokenizer, ProcessorMixin from transformers import PreTrainedTokenizer, ProcessorMixin
from transformers.image_processing_utils import BaseImageProcessor from transformers.image_processing_utils import BaseImageProcessor