aboutsummaryrefslogtreecommitdiffstats
path: root/webui.py
diff options
context:
space:
mode:
authorAUTOMATIC <16777216c@gmail.com>2022-09-11 15:48:36 +0000
committerAUTOMATIC <16777216c@gmail.com>2022-09-11 15:48:36 +0000
commitf194457229e4537912467bc60ac3a873f473a63c (patch)
tree93e9c445a79bb0f2e88e4058734594f318f1151c /webui.py
parent13008bab90305144591e0b2e233214a8b3415bba (diff)
downloadstable-diffusion-webui-gfx803-f194457229e4537912467bc60ac3a873f473a63c.tar.gz
stable-diffusion-webui-gfx803-f194457229e4537912467bc60ac3a873f473a63c.tar.bz2
stable-diffusion-webui-gfx803-f194457229e4537912467bc60ac3a873f473a63c.zip
CLIP interrogator
Diffstat (limited to 'webui.py')
-rw-r--r--webui.py4
1 files changed, 3 insertions, 1 deletions
diff --git a/webui.py b/webui.py
index 70c68338..ca809f79 100644
--- a/webui.py
+++ b/webui.py
@@ -20,7 +20,7 @@ import modules.gfpgan_model
import modules.face_restoration
import modules.realesrgan_model as realesrgan
import modules.esrgan_model as esrgan
-import modules.extras
+import modules.extras
import modules.lowvram
import modules.txt2img
import modules.img2img
@@ -33,6 +33,7 @@ shared.face_restorers.append(modules.face_restoration.FaceRestoration())
esrgan.load_models(cmd_opts.esrgan_models_path)
realesrgan.setup_realesrgan()
+
def load_model_from_config(config, ckpt, verbose=False):
print(f"Loading model from {ckpt}")
pl_sd = torch.load(ckpt, map_location="cpu")
@@ -116,5 +117,6 @@ def webui():
demo.launch(share=cmd_opts.share, server_name="0.0.0.0" if cmd_opts.listen else None, server_port=cmd_opts.port)
+
if __name__ == "__main__":
webui()