fix issue with multi batch size on qwen-image-edit-plus

This commit is contained in:
Jaret Burkett
2025-09-30 09:04:56 -06:00
parent 2e9de5eb50
commit 67ed563e03

View File

@@ -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: