From ea24deefcce4d44339105e0fe6d243743b858074 Mon Sep 17 00:00:00 2001
From: Joshna-Medisetty <joshna.medisetty@intel.com>
Date: Wed, 19 Aug 2026 14:30:12 -0700
Subject: [PATCH] [Bugfix][FLUX.2-klein] Fix duplicated image position ids 
 (#6130)

Signed-off-by: Joshna Medisetty <joshna.medisetty@intel.com>
Signed-off-by: Joshna-Medisetty <joshna.medisetty@intel.com>
Co-authored-by: Cursor <cursoragent@cursor.com>
Co-authored-by: Chendi.Xue <chendi.xue@intel.com>
---
 vllm_omni/diffusion/models/flux2_klein/pipeline_flux2_klein.py | 2 --
 1 file changed, 2 deletions(-)

diff --git a/vllm_omni/diffusion/models/flux2_klein/pipeline_flux2_klein.py b/vllm_omni/diffusion/models/flux2_klein/pipeline_flux2_klein.py
index a4350373b..beb0b3787 100644
--- a/vllm_omni/diffusion/models/flux2_klein/pipeline_flux2_klein.py
+++ b/vllm_omni/diffusion/models/flux2_klein/pipeline_flux2_klein.py
@@ -1020,10 +1020,8 @@ class Flux2KleinPipeline(
 
             if reference_image_latents is not None:
                 latent_model_input = torch.cat([latents, reference_image_latents], dim=1)
-                latent_image_ids = latent_ids
             elif image_latents is not None:
                 latent_model_input = torch.cat([latents, image_latents], dim=1).to(self.transformer.dtype)
-                latent_image_ids = torch.cat([latent_ids, image_latent_ids], dim=1)
 
             positive_kwargs = {
                 "hidden_states": latent_model_input,