mirror of
https://github.com/lllyasviel/stable-diffusion-webui-forge.git
synced 2026-07-21 21:01:24 +08:00
free memory
This commit is contained in:
parent
e6a65a3efd
commit
0850d29559
@ -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
|
||||
|
||||
@ -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()
|
||||
|
||||
Loading…
Reference in New Issue
Block a user