Update sd_samplers_kdiffusion.py

This commit is contained in:
lllyasviel
2024-01-25 08:18:42 -08:00
parent b4997c543f
commit dbdc861d62

View File

@@ -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