fix name

linoytsaban · linoytsaban · commit db54f9d8744d · 2025-05-15T17:50:06.000+03:00
diff --git a/src/diffusers/pipelines/sana/pipeline_sana_sprint_img2img.py b/src/diffusers/pipelines/sana/pipeline_sana_sprint_img2img.py
@@ -627,7 +627,7 @@ def prepare_latents(self,
         )
 
         image = image.to(device=device, dtype=dtype)
-        if image.shape[1] != self.latent_channels:
+        if image.shape[1] != num_channels_latents:
             image_latents = self._encode_vae_image(image=image, generator=generator)
         else:
             image_latents = image

Original file line number	Diff line number	Diff line change
`@@ -627,7 +627,7 @@ def prepare_latents(self,`
`627`	`627`	`)`
`628`	`628`
`629`	`629`	`image = image.to(device=device, dtype=dtype)`
`630`		`- if image.shape[1] != self.latent_channels:`
	`630`	`+ if image.shape[1] != num_channels_latents:`
`631`	`631`	`image_latents = self._encode_vae_image(image=image, generator=generator)`
`632`	`632`	`else:`
`633`	`633`	`image_latents = image`