Skip to content

Commit c984369

Browse files
committed
Fixed resizing of control image resolution for Qwen Image Edit 2509 when using match_target_res
1 parent 42e5e3c commit c984369

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

extensions_built_in/diffusion_models/qwen_image/qwen_image_edit_plus.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -212,7 +212,7 @@ def get_noise_prediction(
212212
control_image_res = VAE_IMAGE_SIZE
213213
if self.model_config.model_kwargs.get("match_target_res", False):
214214
# use the current target size to set the control image res
215-
control_image_res = height * width * self.pipeline.vae_scale_factor
215+
control_image_res = height * self.pipeline.vae_scale_factor * width * self.pipeline.vae_scale_factor
216216

217217
# pack image tokens
218218
latent_model_input = latent_model_input.view(

0 commit comments

Comments
 (0)