Output from decoding latents was changed to a data wrapping object, u…

…nwrap it.
csaluski · Sep 25, 2022 · 7ae0aaa · 7ae0aaa
1 parent 8a111dd
commit 7ae0aaa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py
@@ -366,7 +366,7 @@ def diffuse_from_inits(self, text_embeddings,
 
         # scale and decode the image latents with vae
         latents = 1 / 0.18215 * latents
-        image = self.vae.decode(latents)
+        image = self.vae.decode(latents).sample
 
         image = (image / 2 + 0.5).clamp(0, 1)
         image = image.cpu().permute(0, 2, 3, 1).numpy()