aboutsummaryrefslogtreecommitdiffstats
path: root/webui.py
diff options
context:
space:
mode:
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()