From cba548d8a38e34bcd8ebf842fa75ab0aeae4ebee Mon Sep 17 00:00:00 2001 From: Jianqi Pan Date: Sat, 17 Aug 2024 17:43:42 +0900 Subject: [PATCH] fix(pipeline): k sampler sigmas device (#9189) If Karras is not enabled, a device inconsistency error will occur. This is due to the fact that sigmas were not moved to the specified device. --- .../pipeline_stable_diffusion_k_diffusion.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/pipelines/stable_diffusion_k_diffusion/pipeline_stable_diffusion_k_diffusion.py b/src/diffusers/pipelines/stable_diffusion_k_diffusion/pipeline_stable_diffusion_k_diffusion.py index 1e396cb232..122701ff92 100755 --- a/src/diffusers/pipelines/stable_diffusion_k_diffusion/pipeline_stable_diffusion_k_diffusion.py +++ b/src/diffusers/pipelines/stable_diffusion_k_diffusion/pipeline_stable_diffusion_k_diffusion.py @@ -602,9 +602,9 @@ class StableDiffusionKDiffusionPipeline( 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(device) sigmas = sigmas.to(prompt_embeds.dtype) # 6. Prepare latent variables