Merge pull request #2227 from papuSpartan/master

Refresh list of models/ckpts upon hitting restart gradio in the setti…

2 files changed, 3 insertions, 0 deletions

diff --git a/modules/ui.py b/modules/ui.py
index ee333c3b..f2d16b12 100644
--- a/modules/ui.py
+++ b/modules/ui.py
@@ -1343,6 +1343,7 @@ Requested path was: {f}
             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
index 338f58e1..ca278e94 100644
--- a/webui.py
+++ b/webui.py
@@ -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')