ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-04-30 19:21:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Jaret Burkett
2025-09-30 09:04:56 -06:00
parent 2e9de5eb50
commit 67ed563e03
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
extensions_built_in/diffusion_models/qwen_image/qwen_image_edit_plus.py
View File

@@ -210,9 +210,8 @@ class QwenImageEditPlusModel(QwenImageModel):
img_h2, img_w2 = height // 2, width // 2 img_h2, img_w2 = height // 2, width // 2
img_shapes = [ # build distinct instances per batch item, per mamad8
[(1, img_h2, img_w2)] img_shapes = [(1, img_h2, img_w2) for _ in range(batch_size)]
] * batch_size
# pack controls # pack controls
if batch is None: if batch is None: