Skip to content

S
Stable Diffusion Webui
Commit b34b25b4 authored by AUTOMATIC's avatar AUTOMATIC
Browse files
Options

karras samplers for img2img?

parent 405c8171
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 2 deletions
modules/sd_samplers.py
View file @ b34b25b4
...@@ -338,9 +338,11 @@ class KDiffusionSampler: ...@@ -338,9 +338,11 @@ class KDiffusionSampler:
steps, t_enc = setup_img2img_steps(p, steps) steps, t_enc = setup_img2img_steps(p, steps)
if p.sampler_noise_scheduler_override: if p.sampler_noise_scheduler_override:
sigmas = p.sampler_noise_scheduler_override(steps) sigmas = p.sampler_noise_scheduler_override(steps)
elif self.config is not None and self.config.options.get('scheduler', None) == 'karras':
sigmas = k_diffusion.sampling.get_sigmas_karras(n=steps, sigma_min=0.1, sigma_max=10, device=shared.device)
else: else:
sigmas = self.model_wrap.get_sigmas(steps) sigmas = self.model_wrap.get_sigmas(steps)
noise = noise * sigmas[steps - t_enc - 1] noise = noise * sigmas[steps - t_enc - 1]
xi = x + noise xi = x + noise
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment