diff options
| author | d8ahazard <d8ahazard@gmail.com> | 2022-09-26 15:27:18 +0000 |
|---|---|---|
| committer | d8ahazard <d8ahazard@gmail.com> | 2022-09-26 15:27:18 +0000 |
| commit | 7d5c29b674bacc5654f8613af134632b7cbdb158 (patch) | |
| tree | 08d8d267adf313ea15fdcfeae67f45ff430d4520 /webui.py | |
| parent | 740070ea9cdb254209f66417418f2a4af8b099d6 (diff) | |
| download | stable-diffusion-webui-gfx803-7d5c29b674bacc5654f8613af134632b7cbdb158.tar.gz stable-diffusion-webui-gfx803-7d5c29b674bacc5654f8613af134632b7cbdb158.tar.bz2 stable-diffusion-webui-gfx803-7d5c29b674bacc5654f8613af134632b7cbdb158.zip | |
Cleanup existing directories, fixes
Diffstat (limited to 'webui.py')
| -rw-r--r-- | webui.py | 2 |
1 files changed, 2 insertions, 0 deletions
@@ -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())
|