diff --git a/modules/sd_samplers_cfg_denoiser.py b/modules/sd_samplers_cfg_denoiser.py index 119b3c7c..d92f6949 100644 --- a/modules/sd_samplers_cfg_denoiser.py +++ b/modules/sd_samplers_cfg_denoiser.py @@ -54,7 +54,6 @@ class CFGDenoiser(torch.nn.Module): self.padded_cond_uncond = False self.padded_cond_uncond_v0 = False self.sampler = sampler - self.model_wrap = None self.p = None self.need_last_noise_uncond = False diff --git a/modules/sd_samplers_common.py b/modules/sd_samplers_common.py index 04875048..8f6eded9 100644 --- a/modules/sd_samplers_common.py +++ b/modules/sd_samplers_common.py @@ -180,6 +180,8 @@ def apply_refiner(cfg_denoiser, x, sigma): if not checkpoint_changed: return False + del cfg_denoiser.model_wrap + try: main_entry.refresh_model_loading_parameters() sd_models.forge_model_reload()