From 5ded26cdc70387e7d08144275c138dc0c175a4a2 Mon Sep 17 00:00:00 2001 From: alex choi Date: Mon, 7 Apr 2025 18:59:10 -0400 Subject: [PATCH] ensure dtype match between diffused latents and vae weights (#8391) --- src/diffusers/pipelines/pixart_alpha/pipeline_pixart_sigma.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/pipelines/pixart_alpha/pipeline_pixart_sigma.py b/src/diffusers/pipelines/pixart_alpha/pipeline_pixart_sigma.py index 7f10ee89ee..4b4b85e63e 100644 --- a/src/diffusers/pipelines/pixart_alpha/pipeline_pixart_sigma.py +++ b/src/diffusers/pipelines/pixart_alpha/pipeline_pixart_sigma.py @@ -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: