do not load wait for shared.sd_model to load at startup

1 files changed, 5 insertions, 11 deletions

diff --git a/webui.py b/webui.py
index 357bf4c1..0873a26c 100644
--- a/webui.py
+++ b/webui.py
@@ -6,6 +6,8 @@ import signal
 import re

 import warnings

 import json

+from threading import Thread

+

 from fastapi import FastAPI

 from fastapi.middleware.cors import CORSMiddleware

 from fastapi.middleware.gzip import GZipMiddleware

@@ -191,18 +193,10 @@ def initialize():
     modules.textual_inversion.textual_inversion.list_textual_inversion_templates()

     startup_timer.record("refresh textual inversion templates")

 

-    try:

-        modules.sd_models.load_model()

-    except Exception as e:

-        errors.display(e, "loading stable diffusion model")

-        print("", file=sys.stderr)

-        print("Stable diffusion model failed to load, exiting", file=sys.stderr)

-        exit(1)

-    startup_timer.record("load SD checkpoint")

-

-    shared.opts.data["sd_model_checkpoint"] = shared.sd_model.sd_checkpoint_info.title

+    # load model in parallel to other startup stuff

+    Thread(target=lambda: shared.sd_model).start()

 

-    shared.opts.onchange("sd_model_checkpoint", wrap_queued_call(lambda: modules.sd_models.reload_model_weights()))

+    shared.opts.onchange("sd_model_checkpoint", wrap_queued_call(lambda: modules.sd_models.reload_model_weights()), call=False)

     shared.opts.onchange("sd_vae", wrap_queued_call(lambda: modules.sd_vae.reload_vae_weights()), call=False)

     shared.opts.onchange("sd_vae_as_default", wrap_queued_call(lambda: modules.sd_vae.reload_vae_weights()), call=False)

     shared.opts.onchange("temp_dir", ui_tempdir.on_tmpdir_changed)