custom unet support

1 files changed, 4 insertions, 0 deletions

diff --git a/webui.py b/webui.py
index f9210f41..1e3ff061 100644
--- a/webui.py
+++ b/webui.py
@@ -58,6 +58,7 @@ import modules.sd_hijack
 import modules.sd_hijack_optimizations

 import modules.sd_models

 import modules.sd_vae

+import modules.sd_unet

 import modules.txt2img

 import modules.script_callbacks

 import modules.textual_inversion.textual_inversion

@@ -291,6 +292,9 @@ def initialize_rest(*, reload_script_modules=False):
     modules.sd_hijack.list_optimizers()

     startup_timer.record("scripts list_optimizers")

 

+    modules.sd_unet.list_unets()

+    startup_timer.record("scripts list_unets")

+

     def load_model():

         """

         Accesses shared.sd_model property to load model.