sample_dpmpp_2m_sde no longer crashes when step == 1.

2 years ago · dc198650c0
parent 4796e615dd
commit dc198650c0
1 changed files with 1 additions and 0 deletions
--- a/comfy/k_diffusion/sampling.py
+++ b/comfy/k_diffusion/sampling.py
@ -628,6 +628,7 @@ def sample_dpmpp_2m_sde(model, x, sigmas, extra_args=None, callback=None, disabl
        if sigmas[i + 1] == 0:
            # Denoising step
            x = denoised
+            h = None
        else:
            # DPM-Solver++(2M) SDE
            t, s = -sigmas[i].log(), -sigmas[i + 1].log()