remove redundant op when cfg_pp=False

This commit is contained in:
Balladie 2025-09-14 15:30:45 +09:00
parent f9ca75020e
commit 3de7d3bb25

View File

@ -859,7 +859,9 @@ def sample_dpmpp_2m_sde(model, x, sigmas, extra_args=None, callback=None, disabl
current_denoised = uncond_denoised if cfg_pp else denoised
x = sigmas[i + 1] / sigmas[i] * (-h * eta).exp() * x + alpha_t * (-h_eta).expm1().neg() * denoised
x = x + alpha_t * (-h_eta).exp().neg() * (current_denoised - denoised)
if cfg_pp:
x = x + alpha_t * (-h_eta).exp().neg() * (current_denoised - denoised)
if old_denoised is not None:
r = h_last / h