From 0c9e1c3deb7a43174c0b41234cbdd8a8a5d7ebeb Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Mon, 13 Oct 2025 17:10:46 +0000 Subject: [PATCH] Fixed some fringe cases for qwen image edit. --- .../diffusion_models/qwen_image/qwen_image_edit_plus.py | 8 ++++++++ 1 file changed, 8 insertions(+) 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 14d1e6e..cbd9ad0 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 @@ -165,6 +165,12 @@ class QwenImageEditPlusModel(QwenImageModel): # todo handle not caching text encoder if self.pipeline.text_encoder.device != self.device_torch: self.pipeline.text_encoder.to(self.device_torch) + + if control_images is None: + raise ValueError("Missing control images for QwenImageEditPlusModel") + + if not isinstance(control_images, List): + control_images = [control_images] if control_images is not None and len(control_images) > 0: for i in range(len(control_images)): @@ -200,6 +206,8 @@ class QwenImageEditPlusModel(QwenImageModel): ): with torch.no_grad(): batch_size, num_channels_latents, height, width = latent_model_input.shape + if self.vae.device != self.device_torch: + self.vae.to(self.device_torch) control_image_res = VAE_IMAGE_SIZE if self.model_config.model_kwargs.get("match_target_res", False):