[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:
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,
|
||||
|
||||
Reference in new issue
Block a user