From 67ed563e0348ca02e814040e06b0721f7302adef Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Tue, 30 Sep 2025 09:04:56 -0600 Subject: [PATCH] fix issue with multi batch size on qwen-image-edit-plus --- .../diffusion_models/qwen_image/qwen_image_edit_plus.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) 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 222431f2..95a05c5a 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 @@ -210,9 +210,8 @@ class QwenImageEditPlusModel(QwenImageModel): img_h2, img_w2 = height // 2, width // 2 - img_shapes = [ - [(1, img_h2, img_w2)] - ] * batch_size + # build distinct instances per batch item, per mamad8 + img_shapes = [(1, img_h2, img_w2) for _ in range(batch_size)] # pack controls if batch is None: