fix lora load

handler.py

@@ -24,7 +24,7 @@ if ".dev" in torch.__version__ or "+git" in torch.__version__:
     torch.__long_version__ = torch.__version__
     torch.__version__ = re.search(r'[\d.]+[\d]', torch.__version__).group(0)
 
-from modules import shared, devices, ui_tempdir
+from modules import shared, devices, ui_tempdir, extra_networks, extra_networks_hypernet
 import modules.codeformer_model as codeformer
 import modules.face_restoration
 import modules.gfpgan_model as gfpgan
@@ -95,15 +95,16 @@ def initialize():
     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)
 
-    # shared.reload_hypernetworks()
+    shared.reload_hypernetworks()
 
     # ui_extra_networks.intialize()
     # ui_extra_networks.register_page(ui_extra_networks_textual_inversion.ExtraNetworksPageTextualInversion())
     # ui_extra_networks.register_page(ui_extra_networks_hypernets.ExtraNetworksPageHypernetworks())
     # ui_extra_networks.register_page(ui_extra_networks_checkpoints.ExtraNetworksPageCheckpoints())
 
-    # extra_networks.initialize()
-    # extra_networks.register_extra_network(extra_networks_hypernet.ExtraNetworkHypernet())
+    extra_networks.initialize()
+    extra_networks.register_extra_network(extra_networks_hypernet.ExtraNetworkHypernet())
+    modules.script_callbacks.before_ui_callback()
 
     # if cmd_opts.tls_keyfile is not None and cmd_opts.tls_keyfile is not None:
 

predict.py

@@ -23,7 +23,7 @@ if ".dev" in torch.__version__ or "+git" in torch.__version__:
     torch.__long_version__ = torch.__version__
     torch.__version__ = re.search(r'[\d.]+[\d]', torch.__version__).group(0)
 
-from modules import shared, devices, ui_tempdir
+from modules import shared, devices, ui_tempdir, extra_networks_hypernet, extra_networks
 from modules.api.api import encode_pil_to_base64
 import modules.codeformer_model as codeformer
 import modules.face_restoration
@@ -75,6 +75,10 @@ def initialize():
     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)
 
+    shared.reload_hypernetworks()
+    extra_networks.initialize()
+    extra_networks.register_extra_network(extra_networks_hypernet.ExtraNetworkHypernet())
+    modules.script_callbacks.before_ui_callback()
     # make the program just exit at ctrl+c without waiting for anything
     # def sigint_handler(sig, frame):
     #     print(f'Interrupted with signal {sig} in {frame}')