Refresh list of models/ckpts upon hitting restart gradio in the settings pane

This commit is contained in:
papuSpartan 2022-10-10 19:57:43 -05:00
parent 42bf5fa325
commit 1add3cff84
1 changed files with 4 additions and 0 deletions

View File

@ -39,6 +39,7 @@ 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()
@ -1290,6 +1291,9 @@ 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,
inputs=[],