move list refresh to webui.py and add stdout indicating it's doing so

2 files changed, 2 insertions, 3 deletions

diff --git a/modules/ui.py b/modules/ui.py
index 06ff118f..ae9317a3 100644
--- a/modules/ui.py
+++ b/modules/ui.py
@@ -39,7 +39,6 @@ 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()

@@ -1291,8 +1290,6 @@ 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,

diff --git a/webui.py b/webui.py
index 270584f7..94098c4c 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')