diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2022-10-22 09:23:45 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2022-10-22 09:23:58 +0000 |
| commit | 2b91251637078e04472c91a06a8d9c4db9c1dcf0 (patch) | |
| tree | d2c09617adb426eac493a0976057aefe415745b6 /modules/script_callbacks.py | |
| parent | 26d107374569836161326aae8cd3cc26c1edc372 (diff) | |
| download | stable-diffusion-webui-gfx803-2b91251637078e04472c91a06a8d9c4db9c1dcf0.tar.gz stable-diffusion-webui-gfx803-2b91251637078e04472c91a06a8d9c4db9c1dcf0.tar.bz2 stable-diffusion-webui-gfx803-2b91251637078e04472c91a06a8d9c4db9c1dcf0.zip | |
removed aesthetic gradients as built-in
added support for extensions
Diffstat (limited to 'modules/script_callbacks.py')
| -rw-r--r-- | modules/script_callbacks.py | 42 |
1 files changed, 42 insertions, 0 deletions
diff --git a/modules/script_callbacks.py b/modules/script_callbacks.py new file mode 100644 index 00000000..866b7acd --- /dev/null +++ b/modules/script_callbacks.py @@ -0,0 +1,42 @@ +
+callbacks_model_loaded = []
+callbacks_ui_tabs = []
+
+
+def clear_callbacks():
+ callbacks_model_loaded.clear()
+ callbacks_ui_tabs.clear()
+
+
+def model_loaded_callback(sd_model):
+ for callback in callbacks_model_loaded:
+ callback(sd_model)
+
+
+def ui_tabs_callback():
+ res = []
+
+ for callback in callbacks_ui_tabs:
+ res += callback() or []
+
+ return res
+
+
+def on_model_loaded(callback):
+ """register a function to be called when the stable diffusion model is created; the model is
+ passed as an argument"""
+ callbacks_model_loaded.append(callback)
+
+
+def on_ui_tabs(callback):
+ """register a function to be called when the UI is creating new tabs.
+ The function must either return a None, which means no new tabs to be added, or a list, where
+ each element is a tuple:
+ (gradio_component, title, elem_id)
+
+ gradio_component is a gradio component to be used for contents of the tab (usually gr.Blocks)
+ title is tab text displayed to user in the UI
+ elem_id is HTML id for the tab
+ """
+ callbacks_ui_tabs.append(callback)
+
|