Refresh list of models/ckpts upon hitting restart gradio in the settings pane
This commit is contained in:
parent
42bf5fa325
commit
1add3cff84
1 changed files with 4 additions and 0 deletions
|
@ -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=[],
|
||||
|
|
Loading…
Reference in a new issue