Unverified Commit 5ded26cd authored by alex choi's avatar alex choi Committed by GitHub
Browse files

ensure dtype match between diffused latents and vae weights (#8391)

parent 506f39af
......@@ -868,7 +868,7 @@ class PixArtSigmaPipeline(DiffusionPipeline):
xm.mark_step()
if not output_type == "latent":
image = self.vae.decode(latents / self.vae.config.scaling_factor, return_dict=False)[0]
image = self.vae.decode(latents.to(self.vae.dtype) / self.vae.config.scaling_factor, return_dict=False)[0]
if use_resolution_binning:
image = self.image_processor.resize_and_crop_tensor(image, orig_width, orig_height)
else:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment