Fix memory leak and reduce memory usage

1 files changed, 3 insertions, 0 deletions

diff --git a/webui.py b/webui.py
index c70a11c7..b61a318d 100644
--- a/webui.py
+++ b/webui.py
@@ -22,7 +22,10 @@ import modules.txt2img
 import modules.img2img

 import modules.swinir as swinir

 import modules.sd_models

+from torch.nn.functional import silu

+import ldm

 

+ldm.modules.diffusionmodules.model.nonlinearity = silu

 

 modules.codeformer_model.setup_codeformer()

 modules.gfpgan_model.setup_gfpgan()