Refactor StableDiffusion3Img2ImgPipeline to remove redundant code (#8533)

tolgacangoz · web-flow · commit 9c6e9684a24f · 2024-06-13T07:36:46.000-10:00
diff --git a/src/diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3_img2img.py b/src/diffusers/pipelines/stable_diffusion_3/pipeline_stable_diffusion_3_img2img.py
@@ -591,8 +591,6 @@ def prepare_latents(self, image, timestep, batch_size, num_images_per_prompt, dt
             )
 
         image = image.to(device=device, dtype=dtype)
-        if image.shape[1] == self.vae.config.latent_channels:
-            init_latents = image
 
         batch_size = batch_size * num_images_per_prompt
         if image.shape[1] == self.vae.config.latent_channels:

Original file line number	Diff line number	Diff line change
`@@ -591,8 +591,6 @@ def prepare_latents(self, image, timestep, batch_size, num_images_per_prompt, dt`
`591`	`591`	`)`
`592`	`592`
`593`	`593`	`image = image.to(device=device, dtype=dtype)`
`594`		`- if image.shape[1] == self.vae.config.latent_channels:`
`595`		`- init_latents = image`
`596`	`594`
`597`	`595`	`batch_size = batch_size * num_images_per_prompt`
`598`	`596`	`if image.shape[1] == self.vae.config.latent_channels:`