diff options
| author | brkirch <brkirch@users.noreply.github.com> | 2023-01-04 00:40:16 -0500 |
|---|---|---|
| committer | brkirch <brkirch@users.noreply.github.com> | 2023-01-06 00:14:20 -0500 |
| commit | f6ab5a39d762a7791573d1c52ae5a3024b10e8ed (patch) | |
| tree | c3958d77a6dae42457b571dbe0f1efec7ce45dd2 /modules/sd_samplers.py | |
| parent | d782a95967c9eea753df3333cd1954b6ec73eba0 (diff) | |
| parent | 3e22e294135ed0327ce9d9738655ff03c53df3c0 (diff) | |
Merge branch 'AUTOMATIC1111:master' into sub-quad_attn_opt
Diffstat (limited to 'modules/sd_samplers.py')
| -rw-r--r-- | modules/sd_samplers.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/modules/sd_samplers.py b/modules/sd_samplers.py index 177b5338..e904d860 100644 --- a/modules/sd_samplers.py +++ b/modules/sd_samplers.py @@ -465,7 +465,9 @@ class KDiffusionSampler: if p.sampler_noise_scheduler_override:
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)
+ 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 = k_diffusion.sampling.get_sigmas_karras(n=steps, sigma_min=sigma_min, sigma_max=sigma_max, device=shared.device)
else:
sigmas = self.model_wrap.get_sigmas(steps)
|