xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-11-11 23:46:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

my bad

Browse Source
This commit is contained in:
yoinked 2024-06-26 17:35:17 -07:00 committed by GitHub
parent 7abe28ac3b
commit e5d32c41c0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
comfy/k_diffusion/sampling.py
View File

@ -996,7 +996,8 @@ def sample_euler_ancestral_pp(model, x, sigmas, extra_args=None, callback=None,
dt = sigma_down - sigmas[i] dt = sigma_down - sigmas[i]
x = x + d * dt x = x + d * dt
if sigmas[i + 1] > 0: if sigmas[i + 1] > 0:
x = x + noise_sampler(sigmas[i], sigmas[i + 1]) * s_noise * sigma_up
return x
@torch.no_grad() @torch.no_grad()
def sample_dpmpp_2s_ancestral_pp(model, x, sigmas, extra_args=None, callback=None, disable=None, eta=1., s_noise=1., noise_sampler=None): def sample_dpmpp_2s_ancestral_pp(model, x, sigmas, extra_args=None, callback=None, disable=None, eta=1., s_noise=1., noise_sampler=None):
"""Ancestral sampling with DPM-Solver++(2S) second-order steps.""" """Ancestral sampling with DPM-Solver++(2S) second-order steps."""