@@ -46,7 +46,7 @@ def __init__(self, x, image_cond, sigma, sampling_step, total_sampling_steps):
46
46
47
47
48
48
ScriptCallback = namedtuple ("ScriptCallback" , ["script" , "callback" ])
49
- __callback_map = dict (
49
+ _callback_map = dict (
50
50
callbacks_app_started = [],
51
51
callbacks_model_loaded = [],
52
52
callbacks_ui_tabs = [],
@@ -58,19 +58,19 @@ def __init__(self, x, image_cond, sigma, sampling_step, total_sampling_steps):
58
58
59
59
60
60
def clear_callbacks ():
61
- for callback_list in __callback_map .values ():
61
+ for callback_list in _callback_map .values ():
62
62
callback_list .clear ()
63
63
64
64
def app_started_callback (demo : Optional [Blocks ], app : FastAPI ):
65
- for c in __callback_map ['callbacks_app_started' ]:
65
+ for c in _callback_map ['callbacks_app_started' ]:
66
66
try :
67
67
c .callback (demo , app )
68
68
except Exception :
69
69
report_exception (c , 'app_started_callback' )
70
70
71
71
72
72
def model_loaded_callback (sd_model ):
73
- for c in __callback_map ['callbacks_model_loaded' ]:
73
+ for c in _callback_map ['callbacks_model_loaded' ]:
74
74
try :
75
75
c .callback (sd_model )
76
76
except Exception :
@@ -80,7 +80,7 @@ def model_loaded_callback(sd_model):
80
80
def ui_tabs_callback ():
81
81
res = []
82
82
83
- for c in __callback_map ['callbacks_ui_tabs' ]:
83
+ for c in _callback_map ['callbacks_ui_tabs' ]:
84
84
try :
85
85
res += c .callback () or []
86
86
except Exception :
@@ -90,31 +90,31 @@ def ui_tabs_callback():
90
90
91
91
92
92
def ui_settings_callback ():
93
- for c in __callback_map ['callbacks_ui_settings' ]:
93
+ for c in _callback_map ['callbacks_ui_settings' ]:
94
94
try :
95
95
c .callback ()
96
96
except Exception :
97
97
report_exception (c , 'ui_settings_callback' )
98
98
99
99
100
100
def before_image_saved_callback (params : ImageSaveParams ):
101
- for c in __callback_map ['callbacks_before_image_saved' ]:
101
+ for c in _callback_map ['callbacks_before_image_saved' ]:
102
102
try :
103
103
c .callback (params )
104
104
except Exception :
105
105
report_exception (c , 'before_image_saved_callback' )
106
106
107
107
108
108
def image_saved_callback (params : ImageSaveParams ):
109
- for c in __callback_map ['callbacks_image_saved' ]:
109
+ for c in _callback_map ['callbacks_image_saved' ]:
110
110
try :
111
111
c .callback (params )
112
112
except Exception :
113
113
report_exception (c , 'image_saved_callback' )
114
114
115
115
116
116
def cfg_denoiser_callback (params : CFGDenoiserParams ):
117
- for c in __callback_map ['callbacks_cfg_denoiser' ]:
117
+ for c in _callback_map ['callbacks_cfg_denoiser' ]:
118
118
try :
119
119
c .callback (params )
120
120
except Exception :
@@ -133,27 +133,27 @@ def remove_current_script_callbacks():
133
133
filename = stack [0 ].filename if len (stack ) > 0 else 'unknown file'
134
134
if filename == 'unknown file' :
135
135
return
136
- for callback_list in __callback_map .values ():
136
+ for callback_list in _callback_map .values ():
137
137
for callback_to_remove in [cb for cb in callback_list if cb .script == filename ]:
138
138
callback_list .remove (callback_to_remove )
139
139
140
140
141
141
def remove_callbacks_for_function (callback_func ):
142
- for callback_list in __callback_map .values ():
142
+ for callback_list in _callback_map .values ():
143
143
for callback_to_remove in [cb for cb in callback_list if cb .callback == callback_func ]:
144
144
callback_list .remove (callback_to_remove )
145
145
146
146
147
147
def on_app_started (callback ):
148
148
"""register a function to be called when the webui started, the gradio `Block` component and
149
149
fastapi `FastAPI` object are passed as the arguments"""
150
- add_callback (__callback_map ['callbacks_app_started' ], callback )
150
+ add_callback (_callback_map ['callbacks_app_started' ], callback )
151
151
152
152
153
153
def on_model_loaded (callback ):
154
154
"""register a function to be called when the stable diffusion model is created; the model is
155
155
passed as an argument"""
156
- add_callback (__callback_map ['callbacks_model_loaded' ], callback )
156
+ add_callback (_callback_map ['callbacks_model_loaded' ], callback )
157
157
158
158
159
159
def on_ui_tabs (callback ):
@@ -166,35 +166,34 @@ def on_ui_tabs(callback):
166
166
title is tab text displayed to user in the UI
167
167
elem_id is HTML id for the tab
168
168
"""
169
- add_callback (__callback_map ['callbacks_ui_tabs' ], callback )
169
+ add_callback (_callback_map ['callbacks_ui_tabs' ], callback )
170
170
171
171
172
172
def on_ui_settings (callback ):
173
173
"""register a function to be called before UI settings are populated; add your settings
174
174
by using shared.opts.add_option(shared.OptionInfo(...)) """
175
- add_callback (__callback_map ['callbacks_ui_settings' ], callback )
175
+ add_callback (_callback_map ['callbacks_ui_settings' ], callback )
176
176
177
177
178
178
def on_before_image_saved (callback ):
179
179
"""register a function to be called before an image is saved to a file.
180
180
The callback is called with one argument:
181
181
- params: ImageSaveParams - parameters the image is to be saved with. You can change fields in this object.
182
182
"""
183
- add_callback (__callback_map ['callbacks_before_image_saved' ], callback )
183
+ add_callback (_callback_map ['callbacks_before_image_saved' ], callback )
184
184
185
185
186
186
def on_image_saved (callback ):
187
187
"""register a function to be called after an image is saved to a file.
188
188
The callback is called with one argument:
189
189
- params: ImageSaveParams - parameters the image was saved with. Changing fields in this object does nothing.
190
190
"""
191
- add_callback (__callback_map ['callbacks_image_saved' ], callback )
191
+ add_callback (_callback_map ['callbacks_image_saved' ], callback )
192
192
193
193
194
194
def on_cfg_denoiser (callback ):
195
195
"""register a function to be called in the kdiffussion cfg_denoiser method after building the inner model inputs.
196
196
The callback is called with one argument:
197
197
- params: CFGDenoiserParams - parameters to be passed to the inner model and sampling state details.
198
198
"""
199
- add_callback (__callback_map ['callbacks_cfg_denoiser' ], callback )
200
-
199
+ add_callback (_callback_map ['callbacks_cfg_denoiser' ], callback )
0 commit comments