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

[data] fix internvl plugin when using PIL images (#8129)

Browse Source
This commit is contained in:
Kingsley 2025-05-22 01:32:59 +08:00 committed by GitHub
parent d2a3036a23
commit f08b748199
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -525,14 +525,14 @@ class InternVLPlugin(BasePlugin):
mm_inputs = {} mm_inputs = {}
image_video_patches = [] image_video_patches = []
if len(images) != 0 and isinstance(images[0], str): if len(images) != 0:
images = self._regularize_images( images = self._regularize_images(
images, images,
image_max_pixels=getattr(processor, "image_max_pixels", 1024 * 1024), image_max_pixels=getattr(processor, "image_max_pixels", 1024 * 1024),
image_min_pixels=getattr(processor, "image_min_pixels", 32 * 32), image_min_pixels=getattr(processor, "image_min_pixels", 32 * 32),
)["images"] )["images"]
if len(videos) != 0 and isinstance(videos[0], str): if len(videos) != 0:
videos = self._regularize_videos( videos = self._regularize_videos(
videos, videos,
image_max_pixels=getattr(processor, "video_max_pixels", 256 * 256), image_max_pixels=getattr(processor, "video_max_pixels", 256 * 256),