From dc198650c0d2d281c9d87b23a8917b457a94d837 Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Sun, 21 May 2023 11:34:29 -0400
Subject: [PATCH] sample_dpmpp_2m_sde no longer crashes when step == 1.

---
 comfy/k_diffusion/sampling.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/comfy/k_diffusion/sampling.py b/comfy/k_diffusion/sampling.py
index 94d7a57..c540d74 100644
--- 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()