custom unet support

2023-05-27 15:47:33 +03:00
parent a6e653be26
commit 339b531570
8 changed files with 148 additions and 8 deletions
@@ -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.