diff options
| author | Kohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com> | 2023-05-22 16:29:38 +0000 |
|---|---|---|
| committer | Kohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com> | 2023-05-22 16:29:38 +0000 |
| commit | 403b304162b670597f20b01f147bb042eb78ee5c (patch) | |
| tree | 5bdaafa17d237471b4d6f380793902323e7283e6 /modules/sd_samplers_kdiffusion.py | |
| parent | 65a87ccc9bf92a0fd24a453e2837dd2d19bbf5ce (diff) | |
| download | stable-diffusion-webui-gfx803-403b304162b670597f20b01f147bb042eb78ee5c.tar.gz stable-diffusion-webui-gfx803-403b304162b670597f20b01f147bb042eb78ee5c.tar.bz2 stable-diffusion-webui-gfx803-403b304162b670597f20b01f147bb042eb78ee5c.zip | |
use sigma_max/min in model if sigma_max/min is 0
Diffstat (limited to 'modules/sd_samplers_kdiffusion.py')
| -rw-r--r-- | modules/sd_samplers_kdiffusion.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/modules/sd_samplers_kdiffusion.py b/modules/sd_samplers_kdiffusion.py index e2f18b54..7364ed44 100644 --- a/modules/sd_samplers_kdiffusion.py +++ b/modules/sd_samplers_kdiffusion.py @@ -318,10 +318,11 @@ class KDiffusionSampler: if p.sampler_noise_scheduler_override:
sigmas = p.sampler_noise_scheduler_override(steps)
elif p.enable_custom_k_sched:
+ sigma_min, sigma_max = (0.1, 10) if opts.use_old_karras_scheduler_sigmas else (self.model_wrap.sigmas[0].item(), self.model_wrap.sigmas[-1].item())
sigmas_func = k_diffusion_scheduler[p.k_sched_type]
sigmas_kwargs = {
- 'sigma_min': p.sigma_min,
- 'sigma_max': p.sigma_max
+ 'sigma_min': p.sigma_min or sigma_min,
+ 'sigma_max': p.sigma_max or sigma_max
}
if p.k_sched_type != 'exponential':
sigmas_kwargs['rho'] = p.rho
|