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 32fcaa70..194ea479 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 @@ -247,6 +247,9 @@ class QwenImageEditPlusModel(QwenImageModel): control_img, size=(c_height, c_width), mode="bilinear" ) + # scale to -1 to 1 + control_img = control_img * 2 - 1 + control_latent = self.encode_images( control_img, device=self.device_torch,