From c938679de7b87b4f14894d9f57fe0f40dd6e3c06 Mon Sep 17 00:00:00 2001
From: Jairo Correa <jn.j41r0@gmail.com>
Date: Wed, 28 Sep 2022 22:14:13 -0300
Subject: Fix memory leak and reduce memory usage

---
 webui.py | 3 +++
 1 file changed, 3 insertions(+)

(limited to 'webui.py')

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()
-- 
cgit v1.2.1