From b4e7e042e06190b061c5dbd86ec90b448ccd5425 Mon Sep 17 00:00:00 2001 From: ssit Date: Fri, 30 Jun 2023 15:22:19 -0400 Subject: [PATCH] Fixed sampler not being restored if interrupted --- restart_sampling.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/restart_sampling.py b/restart_sampling.py index 78d31bd..14c1349 100644 --- a/restart_sampling.py +++ b/restart_sampling.py @@ -58,7 +58,9 @@ def restart_sampling(model, seed, steps, cfg, sampler_name, scheduler, positive, return x setattr(k_diffusion_sampling, sample_func_name, restart_wrapper) - samples = common_ksampler(model, seed, steps, cfg, sampler_name, scheduler, - positive, negative, latent_image, denoise=denoise) - setattr(k_diffusion_sampling, sample_func_name, sampler) + try: + samples = common_ksampler(model, seed, steps, cfg, sampler_name, scheduler, + positive, negative, latent_image, denoise=denoise) + finally: + setattr(k_diffusion_sampling, sample_func_name, sampler) return samples