Skip to content

Commit

Permalink
display correct timings after restarting UI
Browse files Browse the repository at this point in the history
  • Loading branch information
AUTOMATIC1111 committed Mar 12, 2023
1 parent a00cd8b commit dfeee78
Show file tree
Hide file tree
Showing 2 changed files with 15 additions and 0 deletions.
3 changes: 3 additions & 0 deletions modules/timer.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,3 +33,6 @@ def summary(self):
res += ")"

return res

def reset(self):
self.__init__()
12 changes: 12 additions & 0 deletions webui.py
Original file line number Diff line number Diff line change
Expand Up @@ -290,24 +290,35 @@ def webui():
wait_on_server(shared.demo)
print('Restarting UI...')

startup_timer.reset()

sd_samplers.set_samplers()

modules.script_callbacks.script_unloaded_callback()
extensions.list_extensions()
startup_timer.record("list extensions")

localization.list_localizations(cmd_opts.localizations_dir)

modelloader.forbid_loaded_nonbuiltin_upscalers()
modules.scripts.reload_scripts()
startup_timer.record("load scripts")

modules.script_callbacks.model_loaded_callback(shared.sd_model)
startup_timer.record("model loaded callback")

modelloader.load_upscalers()
startup_timer.record("load upscalers")

for module in [module for name, module in sys.modules.items() if name.startswith("modules.ui")]:
importlib.reload(module)
startup_timer.record("reload script modules")

modules.sd_models.list_models()
startup_timer.record("list SD models")

shared.reload_hypernetworks()
startup_timer.record("reload hypernetworks")

ui_extra_networks.intialize()
ui_extra_networks.register_page(ui_extra_networks_textual_inversion.ExtraNetworksPageTextualInversion())
Expand All @@ -316,6 +327,7 @@ def webui():

extra_networks.initialize()
extra_networks.register_extra_network(extra_networks_hypernet.ExtraNetworkHypernet())
startup_timer.record("initialize extra networks")


if __name__ == "__main__":
Expand Down

0 comments on commit dfeee78

Please sign in to comment.