From 66722dbea7a2cf50106b03ca3531503a64a3c496 Mon Sep 17 00:00:00 2001 From: YiYi Xu Date: Wed, 31 Jan 2024 09:29:15 -1000 Subject: [PATCH] [sdxl k-diffusion pipeline]move sigma to device (#6757) move sigma to device Co-authored-by: yiyixuxu Co-authored-by: Sayak Paul --- .../pipeline_stable_diffusion_xl_k_diffusion.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/diffusers/pipelines/stable_diffusion_k_diffusion/pipeline_stable_diffusion_xl_k_diffusion.py b/src/diffusers/pipelines/stable_diffusion_k_diffusion/pipeline_stable_diffusion_xl_k_diffusion.py index 39f7f6f7a4..1855ff9672 100644 --- a/src/diffusers/pipelines/stable_diffusion_k_diffusion/pipeline_stable_diffusion_xl_k_diffusion.py +++ b/src/diffusers/pipelines/stable_diffusion_k_diffusion/pipeline_stable_diffusion_xl_k_diffusion.py @@ -888,10 +888,9 @@ class StableDiffusionXLKDiffusionPipeline( sigma_min: float = self.k_diffusion_model.sigmas[0].item() sigma_max: float = self.k_diffusion_model.sigmas[-1].item() sigmas = get_sigmas_karras(n=num_inference_steps, sigma_min=sigma_min, sigma_max=sigma_max) - sigmas = sigmas.to(device) else: sigmas = self.scheduler.sigmas - sigmas = sigmas.to(prompt_embeds.dtype) + sigmas = sigmas.to(dtype=prompt_embeds.dtype, device=device) # 6. Prepare latent variables num_channels_latents = self.unet.config.in_channels