StableDiffusion/WebUI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

remove unwanted changes from #8789

Browse Source
This commit is contained in:
AUTOMATIC 2023-04-29 19:05:43 +03:00
parent f96e6fbd0c
commit a95dc02535
2 changed files with 5 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
modules/script_callbacks.py
View File

@ -94,7 +94,6 @@ callback_map = dict(
callbacks_script_unloaded=[], callbacks_script_unloaded=[],
callbacks_before_ui=[], callbacks_before_ui=[],
callbacks_on_reload=[], callbacks_on_reload=[],
callbacks_on_polling=[],
) )
@ -102,6 +101,7 @@ def clear_callbacks():
for callback_list in callback_map.values(): for callback_list in callback_map.values():
callback_list.clear() callback_list.clear()
def app_started_callback(demo: Optional[Blocks], app: FastAPI): 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: try:
@ -109,14 +109,8 @@ def app_started_callback(demo: Optional[Blocks], app: FastAPI):
except Exception: except Exception:
report_exception(c, 'app_started_callback') 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']: for c in callback_map['callbacks_on_reload']:
try: try:
c.callback() c.callback()
@ -269,14 +263,11 @@ def on_app_started(callback):
add_callback(callback_map['callbacks_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): def on_before_reload(callback):
"""register a function to be called just before the server reloads.""" """register a function to be called just before the server reloads."""
add_callback(callback_map['callbacks_on_reload'], callback) add_callback(callback_map['callbacks_on_reload'], callback)
def on_model_loaded(callback): def on_model_loaded(callback):
"""register a function to be called when the stable diffusion model is created; the model is """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. """ passed as an argument; this function is also called when the script is reloaded. """

4
webui.py
View File

@ -264,13 +264,13 @@ def create_api(app):
def wait_on_server(demo=None): def wait_on_server(demo=None):
while 1: while 1:
time.sleep(0.5) time.sleep(0.5)
modules.script_callbacks.app_polling_callback(None, demo)
if shared.state.need_restart: if shared.state.need_restart:
modules.script_callbacks.app_reload_callback(None, demo)
shared.state.need_restart = False shared.state.need_restart = False
time.sleep(0.5) time.sleep(0.5)
demo.close() demo.close()
time.sleep(0.5) time.sleep(0.5)
modules.script_callbacks.app_reload_callback()
break break