|
|
|
@ -31,12 +31,7 @@ from modules.paths import script_path
|
|
|
|
from modules.shared import cmd_opts
|
|
|
|
from modules.shared import cmd_opts
|
|
|
|
import modules.hypernetworks.hypernetwork
|
|
|
|
import modules.hypernetworks.hypernetwork
|
|
|
|
|
|
|
|
|
|
|
|
modelloader.cleanup_models()
|
|
|
|
|
|
|
|
modules.sd_models.setup_model()
|
|
|
|
|
|
|
|
codeformer.setup_model(cmd_opts.codeformer_models_path)
|
|
|
|
|
|
|
|
gfpgan.setup_model(cmd_opts.gfpgan_models_path)
|
|
|
|
|
|
|
|
shared.face_restorers.append(modules.face_restoration.FaceRestoration())
|
|
|
|
|
|
|
|
modelloader.load_upscalers()
|
|
|
|
|
|
|
|
queue_lock = threading.Lock()
|
|
|
|
queue_lock = threading.Lock()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@ -78,11 +73,18 @@ def wrap_gradio_gpu_call(func, extra_outputs=None):
|
|
|
|
return modules.ui.wrap_gradio_call(f, extra_outputs=extra_outputs)
|
|
|
|
return modules.ui.wrap_gradio_call(f, extra_outputs=extra_outputs)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def initialize():
|
|
|
|
|
|
|
|
modelloader.cleanup_models()
|
|
|
|
|
|
|
|
modules.sd_models.setup_model()
|
|
|
|
|
|
|
|
codeformer.setup_model(cmd_opts.codeformer_models_path)
|
|
|
|
|
|
|
|
gfpgan.setup_model(cmd_opts.gfpgan_models_path)
|
|
|
|
|
|
|
|
shared.face_restorers.append(modules.face_restoration.FaceRestoration())
|
|
|
|
|
|
|
|
modelloader.load_upscalers()
|
|
|
|
|
|
|
|
|
|
|
|
modules.scripts.load_scripts(os.path.join(script_path, "scripts"))
|
|
|
|
modules.scripts.load_scripts(os.path.join(script_path, "scripts"))
|
|
|
|
|
|
|
|
|
|
|
|
shared.sd_model = modules.sd_models.load_model()
|
|
|
|
shared.sd_model = modules.sd_models.load_model()
|
|
|
|
shared.opts.onchange("sd_model_checkpoint", wrap_queued_call(lambda: modules.sd_models.reload_model_weights(shared.sd_model)))
|
|
|
|
shared.opts.onchange("sd_model_checkpoint", wrap_queued_call(lambda: modules.sd_models.reload_model_weights(shared.sd_model)))
|
|
|
|
|
|
|
|
|
|
|
|
shared.opts.onchange("sd_hypernetwork", wrap_queued_call(lambda: modules.hypernetworks.hypernetwork.load_hypernetwork(shared.opts.sd_hypernetwork)))
|
|
|
|
shared.opts.onchange("sd_hypernetwork", wrap_queued_call(lambda: modules.hypernetworks.hypernetwork.load_hypernetwork(shared.opts.sd_hypernetwork)))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@ -129,6 +131,6 @@ def webui():
|
|
|
|
print('Restarting Gradio')
|
|
|
|
print('Restarting Gradio')
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
if __name__ == "__main__":
|
|
|
|
|
|
|
|
initialize()
|
|
|
|
webui()
|
|
|
|
webui()
|
|
|
|
|