From 4ef5cbe5bc78a5cece56d689d8ae4a1f8e446809 Mon Sep 17 00:00:00 2001 From: abionda-sc Date: Sat, 28 Mar 2026 19:00:32 +0000 Subject: [PATCH] Fixing bug where width and height are inverted for control image resizing (#707) --- extensions_built_in/diffusion_models/flux2/flux2_model.py | 4 ++-- .../diffusion_models/qwen_image/qwen_image_edit.py | 4 ++-- .../diffusion_models/qwen_image/qwen_image_edit_plus.py | 8 ++++---- 3 files changed, 8 insertions(+), 8 deletions(-) diff --git a/extensions_built_in/diffusion_models/flux2/flux2_model.py b/extensions_built_in/diffusion_models/flux2/flux2_model.py index 83eb7ca1..1e774746 100644 --- a/extensions_built_in/diffusion_models/flux2/flux2_model.py +++ b/extensions_built_in/diffusion_models/flux2/flux2_model.py @@ -377,8 +377,8 @@ class Flux2Model(BaseModel): "match_target_res", False ): ratio = control_img.shape[2] / control_img.shape[3] - c_width = math.sqrt(control_image_res * ratio) - c_height = c_width / ratio + c_height = math.sqrt(control_image_res * ratio) + c_width = c_height / ratio c_width = round(c_width / 32) * 32 c_height = round(c_height / 32) * 32 diff --git a/extensions_built_in/diffusion_models/qwen_image/qwen_image_edit.py b/extensions_built_in/diffusion_models/qwen_image/qwen_image_edit.py index 724f9e7b..6061889f 100644 --- a/extensions_built_in/diffusion_models/qwen_image/qwen_image_edit.py +++ b/extensions_built_in/diffusion_models/qwen_image/qwen_image_edit.py @@ -181,8 +181,8 @@ class QwenImageEditModel(QwenImageModel): # images are always run through at 1MP, based on diffusers inference code. target_area = 1024 * 1024 ratio = control_images.shape[2] / control_images.shape[3] - width = math.sqrt(target_area * ratio) - height = width / ratio + height = math.sqrt(target_area * ratio) + width = height / ratio width = round(width / 32) * 32 height = round(height / 32) * 32 diff --git a/extensions_built_in/diffusion_models/qwen_image/qwen_image_edit_plus.py b/extensions_built_in/diffusion_models/qwen_image/qwen_image_edit_plus.py index 74a6e293..8e366d03 100644 --- a/extensions_built_in/diffusion_models/qwen_image/qwen_image_edit_plus.py +++ b/extensions_built_in/diffusion_models/qwen_image/qwen_image_edit_plus.py @@ -176,8 +176,8 @@ class QwenImageEditPlusModel(QwenImageModel): for i in range(len(control_images)): # control images are 0 - 1 scale, shape (bs, ch, height, width) ratio = control_images[i].shape[2] / control_images[i].shape[3] - width = math.sqrt(CONDITION_IMAGE_SIZE * ratio) - height = width / ratio + height = math.sqrt(CONDITION_IMAGE_SIZE * ratio) + width = height / ratio width = round(width / 32) * 32 height = round(height / 32) * 32 @@ -264,8 +264,8 @@ class QwenImageEditPlusModel(QwenImageModel): if len(control_img.shape) == 3: control_img = control_img.unsqueeze(0) ratio = control_img.shape[2] / control_img.shape[3] - c_width = math.sqrt(control_image_res * ratio) - c_height = c_width / ratio + c_height = math.sqrt(control_image_res * ratio) + c_width = c_height / ratio c_width = round(c_width / 32) * 32 c_height = round(c_height / 32) * 32