Fix VRAM Issue by only loading in hypernetwork when selected in settings

This commit is contained in:
Fampai
2022-10-08 22:37:35 -04:00
committed by AUTOMATIC1111
parent e00b4df7c6
commit 122d42687b
4 changed files with 25 additions and 18 deletions

View File

@@ -82,6 +82,9 @@ modules.scripts.load_scripts(os.path.join(script_path, "scripts"))
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)))
loaded_hypernetwork = modules.hypernetwork.load_hypernetwork(shared.opts.sd_hypernetwork)
shared.opts.onchange("sd_hypernetwork", wrap_queued_call(lambda: modules.hypernetwork.load_hypernetwork(shared.opts.sd_hypernetwork)))
def webui():
# make the program just exit at ctrl+c without waiting for anything