lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-29 10:41:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #1604 from DenOfEquity/smol-sampler-fix

Browse Source 
Update sampling.py
This commit is contained in:
DenOfEquity
2024-08-30 14:49:08 +01:00
committed by GitHub
parent 4c9380c46a b57056b056
commit 1e0e861b0d
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: if eta:
x = x + noise_sampler(sigmas[i], sigmas[i + 1]) * sigmas[i + 1] * (-2 * eta_h).expm1().neg().sqrt() * s_noise 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 old_denoised = denoised
h_last = h
return x return x
@@ -698,8 +699,9 @@ def sample_dpmpp_3m_sde(model, x, sigmas, extra_args=None, callback=None, disabl
if eta: if eta:
x = x + noise_sampler(sigmas[i], sigmas[i + 1]) * sigmas[i + 1] * (-2 * h * eta).expm1().neg().sqrt() * s_noise 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 denoised_1, denoised_2 = denoised, denoised_1
h_1, h_2 = h, h_1
return x return x
@torch.no_grad() @torch.no_grad()
@@ -910,4 +912,4 @@ def sample_deis(model, x, sigmas, extra_args=None, callback=None, disable=None,
else: else:
buffer_model.append(d_cur.detach()) buffer_model.append(d_cur.detach())
return x_next return x_next