From 4b596768734d7e486edcabdc0f500d8600a462ca Mon Sep 17 00:00:00 2001 From: tastelikefeet <58414341+tastelikefeet@users.noreply.github.com> Date: Fri, 11 Oct 2024 14:08:16 +0800 Subject: [PATCH] fix rescale_image (#2223) --- swift/llm/utils/vision_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/swift/llm/utils/vision_utils.py b/swift/llm/utils/vision_utils.py index bb8f77878..5beef38e5 100644 --- a/swift/llm/utils/vision_utils.py +++ b/swift/llm/utils/vision_utils.py @@ -90,7 +90,7 @@ def rescale_image(img: 'PIL.Image.Image', rescale_image: int = -1) -> 'PIL.Image ratio = width / height height_scaled = math.pow(rescale_image / ratio, 0.5) width_scaled = height_scaled * ratio - return T.Resize((int(width_scaled), int(height_scaled)))(img) + return T.Resize((int(height_scaled), int(width_scaled)))(img) _T = TypeVar('_T')