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

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

View File

@ -39,7 +39,6 @@ import modules.generation_parameters_copypaste
from modules import prompt_parser
from modules.images import save_image
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
mimetypes.init()
@ -1291,8 +1290,6 @@ Requested path was: {f}
shared.state.interrupt()
settings_interface.gradio_ref.do_restart = True
# refresh models so that new models/.ckpt's show up on reload
list_models()
restart_gradio.click(
fn=request_restart,

View File

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