Skip to content

Commit 43711dd

Browse files
committed
fix image encoding
1 parent e0b6c6c commit 43711dd

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/diffusers/pipelines/sana/pipeline_sana_sprint_img2img.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -631,6 +631,7 @@ def prepare_latents(self,
631631
pass
632632
else:
633633
image = self.image_processor.preprocess(image, height=height, width=width)
634+
image = image.to(device=device, dtype=self.vae.dtype)
634635

635636
if image.shape[1] != num_channels_latents:
636637
image = self.vae.encode(image).latent

0 commit comments

Comments
 (0)