Cleanup existing directories, fixes

1 files changed, 2 insertions, 0 deletions

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())