ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-04-22 15:29:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

Allow for matching target resolution with control images for Qwen Image Edit 2509

Browse Source
This commit is contained in:
Jaret Burkett
2025-10-10 14:24:27 -06:00
parent 1bc6dee127
commit e9c4d94256
4 changed files with 42 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
extensions_built_in/diffusion_models/qwen_image/qwen_image_edit_plus.py
View File

@@ -200,6 +200,11 @@ class QwenImageEditPlusModel(QwenImageModel):
):
with torch.no_grad():
batch_size, num_channels_latents, height, width = latent_model_input.shape
control_image_res = VAE_IMAGE_SIZE
if self.model_config.model_kwargs.get("match_target_res", False):
# use the current target size to set the control image res
control_image_res = height * width * self.pipeline.vae_scale_factor
# pack image tokens
latent_model_input = latent_model_input.view(
@@ -244,7 +249,7 @@ class QwenImageEditPlusModel(QwenImageModel):
if len(control_img.shape) == 3:
control_img = control_img.unsqueeze(0)
ratio = control_img.shape[2] / control_img.shape[3]
c_width = math.sqrt(VAE_IMAGE_SIZE * ratio)
c_width = math.sqrt(control_image_res * ratio)
c_height = c_width / ratio
c_width = round(c_width / 32) * 32