michael/stable-diffusion-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

display correct timings after restarting UI

Browse source
This commit is contained in:
AUTOMATIC 2023-03-12 21:25:22 +03:00
parent a00cd8b9c1
commit dfeee786f9
2 changed files with 15 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
modules/timer.py
View file

@ -33,3 +33,6 @@ class Timer:
res += ")" res += ")"
return res return res
def reset(self):
self.__init__()

12
webui.py
View file

@ -290,24 +290,35 @@ def webui():
wait_on_server(shared.demo) wait_on_server(shared.demo)
print('Restarting UI...') print('Restarting UI...')
startup_timer.reset()
sd_samplers.set_samplers() sd_samplers.set_samplers()
modules.script_callbacks.script_unloaded_callback() modules.script_callbacks.script_unloaded_callback()
extensions.list_extensions() extensions.list_extensions()
startup_timer.record("list extensions")
localization.list_localizations(cmd_opts.localizations_dir) localization.list_localizations(cmd_opts.localizations_dir)
modelloader.forbid_loaded_nonbuiltin_upscalers() modelloader.forbid_loaded_nonbuiltin_upscalers()
modules.scripts.reload_scripts() modules.scripts.reload_scripts()
startup_timer.record("load scripts")
modules.script_callbacks.model_loaded_callback(shared.sd_model) modules.script_callbacks.model_loaded_callback(shared.sd_model)
startup_timer.record("model loaded callback")
modelloader.load_upscalers() modelloader.load_upscalers()
startup_timer.record("load upscalers")
for module in [module for name, module in sys.modules.items() if name.startswith("modules.ui")]: for module in [module for name, module in sys.modules.items() if name.startswith("modules.ui")]:
importlib.reload(module) importlib.reload(module)
startup_timer.record("reload script modules")
modules.sd_models.list_models() modules.sd_models.list_models()
startup_timer.record("list SD models")
shared.reload_hypernetworks() shared.reload_hypernetworks()
startup_timer.record("reload hypernetworks")
ui_extra_networks.intialize() ui_extra_networks.intialize()
ui_extra_networks.register_page(ui_extra_networks_textual_inversion.ExtraNetworksPageTextualInversion()) ui_extra_networks.register_page(ui_extra_networks_textual_inversion.ExtraNetworksPageTextualInversion())
@ -316,6 +327,7 @@ def webui():
extra_networks.initialize() extra_networks.initialize()
extra_networks.register_extra_network(extra_networks_hypernet.ExtraNetworkHypernet()) extra_networks.register_extra_network(extra_networks_hypernet.ExtraNetworkHypernet())
startup_timer.record("initialize extra networks")
if __name__ == "__main__": if __name__ == "__main__":