aboutsummaryrefslogtreecommitdiff
path: root/modules/script_callbacks.py
diff options
context:
space:
mode:
authorAUTOMATIC <16777216c@gmail.com>2023-04-29 19:05:43 +0300
committerAUTOMATIC <16777216c@gmail.com>2023-04-29 19:05:43 +0300
commita95dc02535ce647efbb3bc66964ab61b6f1446f4 (patch)
treee2035885cb575028691ea8ac9e0a59eae69b491a /modules/script_callbacks.py
parentf96e6fbd0c628d0a326df7ea9c79aa5339e2b6fd (diff)
remove unwanted changes from #8789
Diffstat (limited to 'modules/script_callbacks.py')
-rw-r--r--modules/script_callbacks.py15
1 files changed, 3 insertions, 12 deletions
diff --git a/modules/script_callbacks.py b/modules/script_callbacks.py
index cede1cb0..17109732 100644
--- a/modules/script_callbacks.py
+++ b/modules/script_callbacks.py
@@ -94,7 +94,6 @@ callback_map = dict(
callbacks_script_unloaded=[],
callbacks_before_ui=[],
callbacks_on_reload=[],
- callbacks_on_polling=[],
)
@@ -102,6 +101,7 @@ def clear_callbacks():
for callback_list in callback_map.values():
callback_list.clear()
+
def app_started_callback(demo: Optional[Blocks], app: FastAPI):
for c in callback_map['callbacks_app_started']:
try:
@@ -109,14 +109,8 @@ def app_started_callback(demo: Optional[Blocks], app: FastAPI):
except Exception:
report_exception(c, 'app_started_callback')
-def app_polling_callback(demo: Optional[Blocks], app: FastAPI):
- for c in callback_map['callbacks_on_polling']:
- try:
- c.callback()
- except Exception:
- report_exception(c, 'callbacks_on_polling')
-def app_reload_callback(demo: Optional[Blocks], app: FastAPI):
+def app_reload_callback():
for c in callback_map['callbacks_on_reload']:
try:
c.callback()
@@ -269,14 +263,11 @@ def on_app_started(callback):
add_callback(callback_map['callbacks_app_started'], callback)
-def on_polling(callback):
- """register a function to be called on each polling of the server."""
- add_callback(callback_map['callbacks_on_polling'], callback)
-
def on_before_reload(callback):
"""register a function to be called just before the server reloads."""
add_callback(callback_map['callbacks_on_reload'], callback)
+
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; this function is also called when the script is reloaded. """