From 7d5c29b674bacc5654f8613af134632b7cbdb158 Mon Sep 17 00:00:00 2001 From: d8ahazard Date: Mon, 26 Sep 2022 10:27:18 -0500 Subject: Cleanup existing directories, fixes --- webui.py | 2 ++ 1 file changed, 2 insertions(+) (limited to 'webui.py') diff --git a/webui.py b/webui.py index 7e0b3296..e71a217c 100644 --- a/webui.py +++ b/webui.py @@ -18,9 +18,11 @@ import modules.shared as shared import modules.swinir_model as swinir import modules.txt2img import modules.ui +from modules import modelloader from modules.paths import script_path from modules.shared import cmd_opts +modelloader.cleanup_models() codeformer.setup_model(cmd_opts.codeformer_models_path) gfpgan.setup_model(cmd_opts.gfpgan_models_path) shared.face_restorers.append(modules.face_restoration.FaceRestoration()) -- cgit v1.2.1