lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-01-26 19:09:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update sampling.py

Browse Source 
shuffle lines so DPM++2M SDE and DPM++3M SDE work with steps == 1
This commit is contained in:
DenOfEquity
2024-08-30 14:42:34 +01:00
committed by GitHub
parent 4c9380c46a
commit b57056b056
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
k_diffusion/sampling.py
View File

@@ -648,8 +648,9 @@ def sample_dpmpp_2m_sde(model, x, sigmas, extra_args=None, callback=None, disabl
if eta:
x = x + noise_sampler(sigmas[i], sigmas[i + 1]) * sigmas[i + 1] * (-2 * eta_h).expm1().neg().sqrt() * s_noise
h_last = h
old_denoised = denoised
h_last = h
return x
@@ -698,8 +699,9 @@ def sample_dpmpp_3m_sde(model, x, sigmas, extra_args=None, callback=None, disabl
if eta:
x = x + noise_sampler(sigmas[i], sigmas[i + 1]) * sigmas[i + 1] * (-2 * h * eta).expm1().neg().sqrt() * s_noise
h_1, h_2 = h, h_1
denoised_1, denoised_2 = denoised, denoised_1
h_1, h_2 = h, h_1
return x
@torch.no_grad()
@@ -910,4 +912,4 @@ def sample_deis(model, x, sigmas, extra_args=None, callback=None, disable=None,
else:
buffer_model.append(d_cur.detach())
return x_next
return x_next