From f930348e79762c2ad5dbd26631182ee1c9db7c76 Mon Sep 17 00:00:00 2001 From: Tianyu Guo Date: Fri, 18 Sep 2026 18:07:45 +0800 Subject: [PATCH] Preserve Qwen-Image 2.1 condition latent order (#7766) Signed-off-by: Tianyu Guo --- .../diffusion/models/qwen_image_21/pipeline_qwen_image_21.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vllm_omni/diffusion/models/qwen_image_21/pipeline_qwen_image_21.py b/vllm_omni/diffusion/models/qwen_image_21/pipeline_qwen_image_21.py index 73d57a542..021fb0f8e 100644 --- a/vllm_omni/diffusion/models/qwen_image_21/pipeline_qwen_image_21.py +++ b/vllm_omni/diffusion/models/qwen_image_21/pipeline_qwen_image_21.py @@ -708,7 +708,7 @@ class QwenImage21Pipeline( raise ValueError( f"Cannot duplicate `image` of batch size {encoded.shape[0]} to {batch_size} text prompts." ) - encoded = torch.cat([encoded] * (batch_size // encoded.shape[0]), dim=0) + encoded = encoded.repeat_interleave(batch_size // encoded.shape[0], dim=0) image_latent_height, image_latent_width = encoded.shape[3:] all_image_latents.append( self._pack_latents(