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(