diff options
Diffstat (limited to 'webui.py')
-rw-r--r-- | webui.py | 4 |
1 files changed, 3 insertions, 1 deletions
@@ -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()
|