Lora support!

update readme to reflect some recent changes

1 files changed, 15 insertions, 0 deletions

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)