lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-05-01 11:41:23 +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

6
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
old_denoised = denoised
h_last = h
old_denoised = denoised
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
denoised_1, denoised_2 = denoised, denoised_1
h_1, h_2 = h, h_1
denoised_1, denoised_2 = denoised, denoised_1
return x
@torch.no_grad()