From 0d5c1818439f1321835bea73601d2b01324b145f Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Tue, 30 Dec 2025 16:40:49 +0000 Subject: [PATCH] Fixed issue where the control images would sometimes be ignored on qwen_image_edit_2511 --- .../qwen_image/qwen_image_edit_plus.py | 14 ++++++++------ version.py | 2 +- 2 files changed, 9 insertions(+), 7 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 4969e890..8272ee46 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 @@ -237,14 +237,16 @@ class QwenImageEditPlusModel(QwenImageModel): # split the latents into batch items so we can concat the controls packed_latents_list = torch.chunk(latent_model_input, batch_size, dim=0) packed_latents_with_controls_list = [] + + batch_control_tensor_list = batch.control_tensor_list + if batch_control_tensor_list is None and batch.control_tensor is not None: + batch_control_tensor_list = [] + for b in range(batch_size): + batch_control_tensor_list.append(batch.control_tensor[b : b + 1]) - if batch.control_tensor_list is not None: - if len(batch.control_tensor_list) != batch_size: - raise ValueError( - "Control tensor list length does not match batch size" - ) + if batch_control_tensor_list is not None: b = 0 - for control_tensor_list in batch.control_tensor_list: + for control_tensor_list in batch_control_tensor_list: # control tensor list is a list of tensors for this batch item controls = [] # pack control diff --git a/version.py b/version.py index b3df5001..36e0689d 100644 --- a/version.py +++ b/version.py @@ -1 +1 @@ -VERSION = "0.7.13" +VERSION = "0.7.14"