[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>
This commit is contained in:
authored and GitHub committed 2026-08-19 16:30:12 -05:00
1 parent c3f80502e0
commit ea24deefcc
1 file changed
-2
@@ -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,