Skip to content

GitLab

"src/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "5eeedd9e3336882d598091e191559f67433b6427"
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
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/diffusers/pipelines/pixart_alpha/pipeline_pixart_sigma.py
View file @ 5ded26cd
...@@ -868,7 +868,7 @@ class PixArtSigmaPipeline(DiffusionPipeline): ...@@ -868,7 +868,7 @@ class PixArtSigmaPipeline(DiffusionPipeline):
xm.mark_step() xm.mark_step()
if not output_type == "latent": 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: if use_resolution_binning:
image = self.image_processor.resize_and_crop_tensor(image, orig_width, orig_height) image = self.image_processor.resize_and_crop_tensor(image, orig_width, orig_height)
else: 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