aboutsummaryrefslogtreecommitdiffstats
path: root/webui.py
diff options
context:
space:
mode:
Diffstat (limited to 'webui.py')
-rw-r--r--webui.py3
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()