Merge pull request AUTOMATIC1111#2227 from papuSpartan/master

Refresh list of models/ckpts upon hitting restart gradio in the setti…
9cat · Oct 11, 2022 · f7e86aa · f7e86aa
2 parents e0ee5bf + d01a2d0
commit f7e86aa
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/modules/ui.py b/modules/ui.py
@@ -1343,6 +1343,7 @@ def request_restart():
             shared.state.interrupt()
             settings_interface.gradio_ref.do_restart = True
 
+
         restart_gradio.click(
             fn=request_restart,
             inputs=[],

diff --git a/webui.py b/webui.py
@@ -124,6 +124,8 @@ def sigint_handler(sig, frame):
         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')