Skip to content

Commit e0b6c6c

Browse files
committed
fix image encoding
1 parent 940a7a5 commit e0b6c6c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/pipelines/sana/pipeline_sana_sprint_img2img.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -633,7 +633,7 @@ def prepare_latents(self,
633633
image = self.image_processor.preprocess(image, height=height, width=width)
634634

635635
if image.shape[1] != num_channels_latents:
636-
image = self.vae.encode(image, generator=generator).latent
636+
image = self.vae.encode(image).latent
637637
image_latents = image * self.vae.config.scaling_factor
638638
else:
639639
image_latents = image

0 commit comments

Comments
 (0)