hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-04 03:10:21 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

change the behavior of discard_next_to_last_sigma for sgm_uniform to match other schedulers

Browse Source
This commit is contained in:
AUTOMATIC1111
2024-03-20 10:29:52 +03:00
parent ac9aa44cb8
commit 31306ce672
2 changed files with 1 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
modules/sd_samplers_kdiffusion.py
View File

@@ -113,10 +113,6 @@ class KDiffusionSampler(sd_samplers_common.Sampler):
if scheduler.need_inner_model:
sigmas_kwargs['inner_model'] = self.model_wrap
if scheduler.name == "sgm_uniform": # XXX check this
# Ensure the "step" will be target step + 1
steps += 1 if not discard_next_to_last_sigma else 0
sigmas = scheduler.function(n=steps, **sigmas_kwargs, device=shared.device)
if discard_next_to_last_sigma: