Fix referenced before assignment

For https://github.com/BlenderNeko/ComfyUI_TiledKSampler/issues/13
This commit is contained in:
Alexopus 2023-08-17 22:30:07 +02:00 committed by GitHub
parent be9c5e25bc
commit e59fe0537a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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)