From 855b9e3d1c5a1bd8c2d815d38a38bc7c410be5a8 Mon Sep 17 00:00:00 2001
From: AUTOMATIC <16777216c@gmail.com>
Date: Sat, 21 Jan 2023 16:15:53 +0300
Subject: Lora support! update readme to reflect some recent changes

---
 modules/extra_networks_hypernet.py |  2 +-
 modules/script_callbacks.py        | 15 +++++++++++++++
 modules/ui_extra_networks.py       |  2 +-
 3 files changed, 17 insertions(+), 2 deletions(-)

(limited to 'modules')

diff --git a/modules/extra_networks_hypernet.py b/modules/extra_networks_hypernet.py
index 6a0c4ba8..ff279a1f 100644
--- a/modules/extra_networks_hypernet.py
+++ b/modules/extra_networks_hypernet.py
@@ -17,5 +17,5 @@ class ExtraNetworkHypernet(extra_networks.ExtraNetwork):
 
         hypernetwork.load_hypernetworks(names, multipliers)
 
-    def deactivate(p, self):
+    def deactivate(self, p):
         pass
diff --git a/modules/script_callbacks.py b/modules/script_callbacks.py
index a9e19236..4bb45ec7 100644
--- a/modules/script_callbacks.py
+++ b/modules/script_callbacks.py
@@ -73,6 +73,7 @@ callback_map = dict(
     callbacks_image_grid=[],
     callbacks_infotext_pasted=[],
     callbacks_script_unloaded=[],
+    callbacks_before_ui=[],
 )
 
 
@@ -189,6 +190,14 @@ def script_unloaded_callback():
             report_exception(c, 'script_unloaded')
 
 
+def before_ui_callback():
+    for c in reversed(callback_map['callbacks_before_ui']):
+        try:
+            c.callback()
+        except Exception:
+            report_exception(c, 'before_ui')
+
+
 def add_callback(callbacks, fun):
     stack = [x for x in inspect.stack() if x.filename != __file__]
     filename = stack[0].filename if len(stack) > 0 else 'unknown file'
@@ -313,3 +322,9 @@ def on_script_unloaded(callback):
     the script did should be reverted here"""
 
     add_callback(callback_map['callbacks_script_unloaded'], callback)
+
+
+def on_before_ui(callback):
+    """register a function to be called before the UI is created."""
+
+    add_callback(callback_map['callbacks_before_ui'], callback)
diff --git a/modules/ui_extra_networks.py b/modules/ui_extra_networks.py
index 253e90f7..796e879c 100644
--- a/modules/ui_extra_networks.py
+++ b/modules/ui_extra_networks.py
@@ -10,7 +10,7 @@ extra_pages = []
 
 
 def register_page(page):
-    """registers extra networks page for the UI; recommend doing it in on_app_started() callback for extensions"""
+    """registers extra networks page for the UI; recommend doing it in on_before_ui() callback for extensions"""
 
     extra_pages.append(page)
 
-- 
cgit v1.2.1