diff options
author | DepFA <35278260+dfaker@users.noreply.github.com> | 2022-11-04 16:02:25 +0000 |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-11-04 16:02:25 +0000 |
commit | 5844ef8a9a165e0f456a4658bda830282cf5a55e (patch) | |
tree | 218e8522d321202571730a72f783b51881dd9ac9 /modules | |
parent | c3cd0d7a86f35a5bfc58fdc3ecfaf203c0aee06f (diff) | |
download | stable-diffusion-webui-gfx803-5844ef8a9a165e0f456a4658bda830282cf5a55e.tar.gz stable-diffusion-webui-gfx803-5844ef8a9a165e0f456a4658bda830282cf5a55e.tar.bz2 stable-diffusion-webui-gfx803-5844ef8a9a165e0f456a4658bda830282cf5a55e.zip |
remove private underscore indicator
Diffstat (limited to 'modules')
-rw-r--r-- | modules/script_callbacks.py | 36 |
1 files changed, 18 insertions, 18 deletions
diff --git a/modules/script_callbacks.py b/modules/script_callbacks.py index 83da7ca4..74dfb880 100644 --- a/modules/script_callbacks.py +++ b/modules/script_callbacks.py @@ -46,7 +46,7 @@ class CFGDenoiserParams: ScriptCallback = namedtuple("ScriptCallback", ["script", "callback"])
-_callback_map = dict(
+callback_map = dict(
callbacks_app_started=[],
callbacks_model_loaded=[],
callbacks_ui_tabs=[],
@@ -58,11 +58,11 @@ _callback_map = dict( def clear_callbacks():
- for callback_list in _callback_map.values():
+ 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']:
+ for c in callback_map['callbacks_app_started']:
try:
c.callback(demo, app)
except Exception:
@@ -70,7 +70,7 @@ def app_started_callback(demo: Optional[Blocks], app: FastAPI): def model_loaded_callback(sd_model):
- for c in _callback_map['callbacks_model_loaded']:
+ for c in callback_map['callbacks_model_loaded']:
try:
c.callback(sd_model)
except Exception:
@@ -80,7 +80,7 @@ def model_loaded_callback(sd_model): def ui_tabs_callback():
res = []
- for c in _callback_map['callbacks_ui_tabs']:
+ for c in callback_map['callbacks_ui_tabs']:
try:
res += c.callback() or []
except Exception:
@@ -90,7 +90,7 @@ def ui_tabs_callback(): def ui_settings_callback():
- for c in _callback_map['callbacks_ui_settings']:
+ for c in callback_map['callbacks_ui_settings']:
try:
c.callback()
except Exception:
@@ -98,7 +98,7 @@ def ui_settings_callback(): def before_image_saved_callback(params: ImageSaveParams):
- for c in _callback_map['callbacks_before_image_saved']:
+ for c in callback_map['callbacks_before_image_saved']:
try:
c.callback(params)
except Exception:
@@ -106,7 +106,7 @@ def before_image_saved_callback(params: ImageSaveParams): def image_saved_callback(params: ImageSaveParams):
- for c in _callback_map['callbacks_image_saved']:
+ for c in callback_map['callbacks_image_saved']:
try:
c.callback(params)
except Exception:
@@ -114,7 +114,7 @@ def image_saved_callback(params: ImageSaveParams): def cfg_denoiser_callback(params: CFGDenoiserParams):
- for c in _callback_map['callbacks_cfg_denoiser']:
+ for c in callback_map['callbacks_cfg_denoiser']:
try:
c.callback(params)
except Exception:
@@ -133,13 +133,13 @@ def remove_current_script_callbacks(): filename = stack[0].filename if len(stack) > 0 else 'unknown file'
if filename == 'unknown file':
return
- for callback_list in _callback_map.values():
+ for callback_list in callback_map.values():
for callback_to_remove in [cb for cb in callback_list if cb.script == filename]:
callback_list.remove(callback_to_remove)
def remove_callbacks_for_function(callback_func):
- for callback_list in _callback_map.values():
+ for callback_list in callback_map.values():
for callback_to_remove in [cb for cb in callback_list if cb.callback == callback_func]:
callback_list.remove(callback_to_remove)
@@ -147,13 +147,13 @@ def remove_callbacks_for_function(callback_func): def on_app_started(callback):
"""register a function to be called when the webui started, the gradio `Block` component and
fastapi `FastAPI` object are passed as the arguments"""
- add_callback(_callback_map['callbacks_app_started'], callback)
+ add_callback(callback_map['callbacks_app_started'], 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"""
- add_callback(_callback_map['callbacks_model_loaded'], callback)
+ add_callback(callback_map['callbacks_model_loaded'], callback)
def on_ui_tabs(callback):
@@ -166,13 +166,13 @@ def on_ui_tabs(callback): title is tab text displayed to user in the UI
elem_id is HTML id for the tab
"""
- add_callback(_callback_map['callbacks_ui_tabs'], callback)
+ add_callback(callback_map['callbacks_ui_tabs'], callback)
def on_ui_settings(callback):
"""register a function to be called before UI settings are populated; add your settings
by using shared.opts.add_option(shared.OptionInfo(...)) """
- add_callback(_callback_map['callbacks_ui_settings'], callback)
+ add_callback(callback_map['callbacks_ui_settings'], callback)
def on_before_image_saved(callback):
@@ -180,7 +180,7 @@ def on_before_image_saved(callback): The callback is called with one argument:
- params: ImageSaveParams - parameters the image is to be saved with. You can change fields in this object.
"""
- add_callback(_callback_map['callbacks_before_image_saved'], callback)
+ add_callback(callback_map['callbacks_before_image_saved'], callback)
def on_image_saved(callback):
@@ -188,7 +188,7 @@ def on_image_saved(callback): The callback is called with one argument:
- params: ImageSaveParams - parameters the image was saved with. Changing fields in this object does nothing.
"""
- add_callback(_callback_map['callbacks_image_saved'], callback)
+ add_callback(callback_map['callbacks_image_saved'], callback)
def on_cfg_denoiser(callback):
@@ -196,4 +196,4 @@ def on_cfg_denoiser(callback): The callback is called with one argument:
- params: CFGDenoiserParams - parameters to be passed to the inner model and sampling state details.
"""
- add_callback(_callback_map['callbacks_cfg_denoiser'], callback)
+ add_callback(callback_map['callbacks_cfg_denoiser'], callback)
|