From f194457229e4537912467bc60ac3a873f473a63c Mon Sep 17 00:00:00 2001 From: AUTOMATIC <16777216c@gmail.com> Date: Sun, 11 Sep 2022 18:48:36 +0300 Subject: CLIP interrogator --- webui.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'webui.py') 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() -- cgit v1.2.1