Cleanup existing directories, fixes

This commit is contained in:
d8ahazard
2022-09-26 10:27:18 -05:00
parent 740070ea9c
commit 7d5c29b674
6 changed files with 48 additions and 14 deletions

View File

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