diff options
| author | hentailord85ez <112723046+hentailord85ez@users.noreply.github.com> | 2022-12-26 20:49:13 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-12-26 20:49:13 +0000 |
| commit | 4df5009acb6832daef1ff5949404b5aadc8f8fa4 (patch) | |
| tree | 19175933308e9a94b3c285af16bd65ac5015f58a /modules/sd_samplers.py | |
| parent | 4af3ca5393151d61363c30eef4965e694eeac15e (diff) | |
| download | stable-diffusion-webui-gfx803-4df5009acb6832daef1ff5949404b5aadc8f8fa4.tar.gz stable-diffusion-webui-gfx803-4df5009acb6832daef1ff5949404b5aadc8f8fa4.tar.bz2 stable-diffusion-webui-gfx803-4df5009acb6832daef1ff5949404b5aadc8f8fa4.zip | |
Update sd_samplers.py
Diffstat (limited to 'modules/sd_samplers.py')
| -rw-r--r-- | modules/sd_samplers.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/modules/sd_samplers.py b/modules/sd_samplers.py index 177b5338..f4473832 100644 --- a/modules/sd_samplers.py +++ b/modules/sd_samplers.py @@ -462,6 +462,9 @@ class KDiffusionSampler: return extra_params_kwargs
def get_sigmas(self, p, steps):
+ disc = opts.always_discard_next_to_last_sigma or (self.config is not None and self.config.options.get('discard_next_to_last_sigma', False))
+ steps += 1 if disc else 0
+
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':
@@ -469,7 +472,7 @@ class KDiffusionSampler: else:
sigmas = self.model_wrap.get_sigmas(steps)
- if self.config is not None and self.config.options.get('discard_next_to_last_sigma', False):
+ if disc:
sigmas = torch.cat([sigmas[:-2], sigmas[-1:]])
return sigmas
|