ros/LLaMA-Factory
1
0
Fork 0
Code Issues Pull Requests Actions 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
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