ros/LLaMA-Factory
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[data] fix mm template (#7181)

Browse Source 
Former-commit-id: 648616d473c81d393592806307e3e25b159cb278
This commit is contained in:
hoshi-hiyouga
2025-03-06 15:18:32 +08:00
committed by GitHub
parent 9f16c50155
commit bb8aba5abf
2 changed files with 12 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1042,15 +1042,15 @@ class Qwen2VLPlugin(BasePlugin):
image = super()._preprocess_image(image, **kwargs)
if min(image.width, image.height) < 28:
width, height = max(image.width, 28), max(image.height, 28)
image = image.resize((width, height), resample=Image.Resampling.NEAREST)
image = image.resize((width, height))
if image.width / image.height > 200:
width, height = image.height * 180, image.height
image = image.resize((width, height), resample=Image.Resampling.NEAREST)
image = image.resize((width, height))
if image.height / image.width > 200:
width, height = image.width, image.width * 180
image = image.resize((width, height), resample=Image.Resampling.NEAREST)
image = image.resize((width, height))
return image