add sampler_noise_scheduler_override switch (bc38c80c) · Commits · github_fork / Stable Diffusion Webui

modules/sd_samplers.py

+8 −2

Original line number	Diff line number	Diff line
		@@ -290,6 +290,9 @@ class KDiffusionSampler:
		def sample_img2img(self, p, x, noise, conditioning, unconditional_conditioning, steps=None):
		steps, t_enc = setup_img2img_steps(p, steps)

		if p.sampler_noise_scheduler_override:
		sigmas = p.sampler_noise_scheduler_override(steps)
		else:
		sigmas = self.model_wrap.get_sigmas(steps)

		noise = noise * sigmas[steps - t_enc - 1]
		@@ -306,6 +309,9 @@ class KDiffusionSampler:
		def sample(self, p, x, conditioning, unconditional_conditioning, steps=None):
		steps = steps or p.steps

		if p.sampler_noise_scheduler_override:
		sigmas = p.sampler_noise_scheduler_override(steps)
		else:
		sigmas = self.model_wrap.get_sigmas(steps)
		x = x * sigmas[0]