Skip to content

Commit a88265c

Browse files
authored
prepare_latents is not copied from
1 parent 105f1f7 commit a88265c

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

src/diffusers/pipelines/flux/pipeline_flux_control_inpaint.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -603,7 +603,6 @@ def disable_vae_tiling(self):
603603
"""
604604
self.vae.disable_tiling()
605605

606-
# Copied from diffusers.pipelines.flux.pipeline_flux_img2img.FluxImg2ImgPipeline.prepare_latents
607606
def prepare_latents(
608607
self,
609608
image,
@@ -649,7 +648,7 @@ def prepare_latents(
649648
noise = randn_tensor(shape, generator=generator, device=device, dtype=dtype)
650649
latents = self.scheduler.scale_noise(image_latents, timestep, noise)
651650
latents = self._pack_latents(latents, batch_size, num_channels_latents, height, width)
652-
return latents, latent_image_ids
651+
return latents, noise, image_latents, latent_image_ids
653652

654653
# Copied from diffusers.pipelines.controlnet_sd3.pipeline_stable_diffusion_3_controlnet.StableDiffusion3ControlNetPipeline.prepare_image
655654
def prepare_image(

0 commit comments

Comments
 (0)