From c1a5ad243a4f1644554bee9298183f95463386c7 Mon Sep 17 00:00:00 2001 From: Marcus Wong Date: Mon, 18 Dec 2023 15:49:31 +0800 Subject: [PATCH 1/2] fix RuntimeError: Input type (float) and bias type (c10::Half) should be the same --- .../pipelines/stable_diffusion/pipeline_stable_diffusion.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py index 2ad90f049922..0d5f324cabaa 100644 --- a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py +++ b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py @@ -1031,6 +1031,9 @@ def __call__( callback(step_idx, t, latents) if not output_type == "latent": + # Ensure latents are always the same type as the VAE + latents = latents.to(next(iter(self.vae.post_quant_conv.parameters())).dtype) + image = self.vae.decode(latents / self.vae.config.scaling_factor, return_dict=False, generator=generator)[ 0 ] From 4903a5485de610056704a064279e2198f7143e02 Mon Sep 17 00:00:00 2001 From: Marcus Wong Date: Mon, 18 Dec 2023 16:02:48 +0800 Subject: [PATCH 2/2] W293 [*] Blank line contains whitespace --- .../pipelines/stable_diffusion/pipeline_stable_diffusion.py | 1 - 1 file changed, 1 deletion(-) diff --git a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py index 0d5f324cabaa..9ad389dd26b2 100644 --- a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py +++ b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py @@ -1033,7 +1033,6 @@ def __call__( if not output_type == "latent": # Ensure latents are always the same type as the VAE latents = latents.to(next(iter(self.vae.post_quant_conv.parameters())).dtype) - image = self.vae.decode(latents / self.vae.config.scaling_factor, return_dict=False, generator=generator)[ 0 ]