From dbdc861d62a0c06132a2902abb3e59a6c53c5f10 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Thu, 25 Jan 2024 08:18:42 -0800 Subject: [PATCH] Update sd_samplers_kdiffusion.py --- modules/sd_samplers_kdiffusion.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/sd_samplers_kdiffusion.py b/modules/sd_samplers_kdiffusion.py index 1c1f7f16..1db4903f 100644 --- a/modules/sd_samplers_kdiffusion.py +++ b/modules/sd_samplers_kdiffusion.py @@ -155,7 +155,7 @@ class KDiffusionSampler(sd_samplers_common.Sampler): sigmas = self.get_sigmas(p, steps) sigma_sched = sigmas[steps - t_enc - 1:] - xi = x + noise * sigma_sched[0] + xi = x.to(noise) + noise * sigma_sched[0] if opts.img2img_extra_noise > 0: p.extra_generation_params["Extra noise"] = opts.img2img_extra_noise