From e59fe0537a98b78beb7f65ae0e221c501bbf64e5 Mon Sep 17 00:00:00 2001 From: Alexopus Date: Thu, 17 Aug 2023 22:30:07 +0200 Subject: [PATCH] Fix referenced before assignment For https://github.com/BlenderNeko/ComfyUI_TiledKSampler/issues/13 --- comfy/k_diffusion/sampling.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comfy/k_diffusion/sampling.py b/comfy/k_diffusion/sampling.py index beaa623f..eb088d92 100644 --- a/comfy/k_diffusion/sampling.py +++ b/comfy/k_diffusion/sampling.py @@ -649,7 +649,7 @@ def sample_dpmpp_3m_sde(model, x, sigmas, extra_args=None, callback=None, disabl s_in = x.new_ones([x.shape[0]]) denoised_1, denoised_2 = None, None - h_1, h_2 = None, None + h, h_1, h_2 = None, None, None for i in trange(len(sigmas) - 1, disable=disable): denoised = model(x, sigmas[i] * s_in, **extra_args)