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

1 files changed, 4 insertions, 0 deletions

diff --git a/modules/ui.py b/modules/ui.py
index e8039d76..06ff118f 100644
--- a/modules/ui.py
+++ b/modules/ui.py
@@ -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=[],