From 452914eeaa16ddde0f95dfe3ed1d0607f942a056 Mon Sep 17 00:00:00 2001 From: lewisay Date: Tue, 31 Oct 2023 14:00:29 +0800 Subject: [PATCH] fix "UnboundLocalError: local variable 'h' referenced before assignment" --- k_diffusion/sampling.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/k_diffusion/sampling.py b/k_diffusion/sampling.py index 9f10e8d..f5b5b90 100644 --- a/k_diffusion/sampling.py +++ b/k_diffusion/sampling.py @@ -662,7 +662,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_1, h_2, h = None, None, None for i in trange(len(sigmas) - 1, disable=disable): denoised = model(x, sigmas[i] * s_in, **extra_args)