diff options
| author | Jairo Correa <jn.j41r0@gmail.com> | 2022-09-29 01:14:13 +0000 |
|---|---|---|
| committer | Jairo Correa <jn.j41r0@gmail.com> | 2022-09-29 01:14:13 +0000 |
| commit | c938679de7b87b4f14894d9f57fe0f40dd6e3c06 (patch) | |
| tree | 31119c9dc5d04648bf873069c690b4fd53dc3805 /webui.py | |
| parent | 041d2aefc082c2883aa7e28ee3e4a990b3be9758 (diff) | |
| download | stable-diffusion-webui-gfx803-c938679de7b87b4f14894d9f57fe0f40dd6e3c06.tar.gz stable-diffusion-webui-gfx803-c938679de7b87b4f14894d9f57fe0f40dd6e3c06.tar.bz2 stable-diffusion-webui-gfx803-c938679de7b87b4f14894d9f57fe0f40dd6e3c06.zip | |
Fix memory leak and reduce memory usage
Diffstat (limited to 'webui.py')
| -rw-r--r-- | webui.py | 3 |
1 files changed, 3 insertions, 0 deletions
@@ -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()
|