From 0850d29559be2d855c0400419b1c4461042cbd41 Mon Sep 17 00:00:00 2001 From: Haoming Date: Fri, 3 Oct 2025 14:23:59 +0800 Subject: [PATCH] free memory --- modules/sd_samplers_cfg_denoiser.py | 1 - modules/sd_samplers_common.py | 2 ++ 2 files changed, 2 insertions(+), 1 deletion(-) 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()