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

[misc] update mm plugin (#6691)

Browse Source 
Former-commit-id: c0caa7afc60ed3015fe6c263ba3566202ba934f1
This commit is contained in:
hoshi-hiyouga 2025-01-17 23:04:26 +08:00 committed by GitHub
parent bbf334f823
commit f87c788154
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -91,7 +91,7 @@ class BasePlugin:
if (image.width * image.height) > image_resolution: if (image.width * image.height) > image_resolution:
resize_factor = math.sqrt(image_resolution / (image.width * image.height)) resize_factor = math.sqrt(image_resolution / (image.width * image.height))
width, height = int(image.width * resize_factor), int(image.height * resize_factor) width, height = int(image.width * resize_factor), int(image.height * resize_factor)
image = image.resize((width, height), resample=Image.NEAREST) image = image.resize((width, height), resample=Image.Resampling.NEAREST)
if image.mode != "RGB": if image.mode != "RGB":
image = image.convert("RGB") image = image.convert("RGB")
@ -780,15 +780,15 @@ class Qwen2vlPlugin(BasePlugin):
image = super()._preprocess_image(image, **kwargs) image = super()._preprocess_image(image, **kwargs)
if min(image.width, image.height) < 28: if min(image.width, image.height) < 28:
width, height = max(image.width, 28), max(image.height, 28) width, height = max(image.width, 28), max(image.height, 28)
image = image.resize((width, height), resample=Image.NEAREST) image = image.resize((width, height), resample=Image.Resampling.NEAREST)
if image.width / image.height > 200: if image.width / image.height > 200:
width, height = image.height * 180, image.height width, height = image.height * 180, image.height
image = image.resize((width, height), resample=Image.NEAREST) image = image.resize((width, height), resample=Image.Resampling.NEAREST)
if image.height / image.width > 200: if image.height / image.width > 200:
width, height = image.width, image.width * 180 width, height = image.width, image.width * 180
image = image.resize((width, height), resample=Image.NEAREST) image = image.resize((width, height), resample=Image.Resampling.NEAREST)
return image return image