mirror of
https://github.com/ostris/ai-toolkit.git
synced 2026-01-26 16:39:47 +00:00
fix issue with multi batch size on qwen-image-edit-plus
This commit is contained in:
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user