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

move list refresh to webui.py and add stdout indicating it's doing so

Browse source
This commit is contained in:
papuSpartan 2022-10-11 08:03:31 -05:00
parent 1add3cff84
commit d01a2d0156
2 changed files with 2 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
modules/ui.py
View file

@ -39,7 +39,6 @@ import modules.generation_parameters_copypaste
from modules import prompt_parser from modules import prompt_parser
from modules.images import save_image from modules.images import save_image
import modules.textual_inversion.ui import modules.textual_inversion.ui
from modules.sd_models import list_models
# this is a fix for Windows users. Without it, javascript files will be served with text/html content-type and the browser will not show any UI # this is a fix for Windows users. Without it, javascript files will be served with text/html content-type and the browser will not show any UI
mimetypes.init() mimetypes.init()
@ -1291,8 +1290,6 @@ Requested path was: {f}
shared.state.interrupt() shared.state.interrupt()
settings_interface.gradio_ref.do_restart = True settings_interface.gradio_ref.do_restart = True
# refresh models so that new models/.ckpt's show up on reload
list_models()
restart_gradio.click( restart_gradio.click(
fn=request_restart, fn=request_restart,

2
webui.py
View file

@ -124,6 +124,8 @@ def webui():
modules.scripts.reload_scripts(os.path.join(script_path, "scripts")) modules.scripts.reload_scripts(os.path.join(script_path, "scripts"))
print('Reloading modules: modules.ui') print('Reloading modules: modules.ui')
importlib.reload(modules.ui) importlib.reload(modules.ui)
print('Refreshing Model List')
modules.sd_models.list_models()
print('Restarting Gradio') print('Restarting Gradio')