diff options
| author | papuSpartan <mcgpapu@gmail.com> | 2022-10-11 00:57:43 +0000 |
|---|---|---|
| committer | papuSpartan <mcgpapu@gmail.com> | 2022-10-11 00:57:43 +0000 |
| commit | 1add3cff84b7e2436d69b1e97ae689281e4a7c33 (patch) | |
| tree | 671000f38ad4f42949956da4dd846c500cb188ec | |
| parent | 42bf5fa3256bff5e4640e5a626e750d4e49e01e1 (diff) | |
| download | stable-diffusion-webui-gfx803-1add3cff84b7e2436d69b1e97ae689281e4a7c33.tar.gz stable-diffusion-webui-gfx803-1add3cff84b7e2436d69b1e97ae689281e4a7c33.tar.bz2 stable-diffusion-webui-gfx803-1add3cff84b7e2436d69b1e97ae689281e4a7c33.zip | |
Refresh list of models/ckpts upon hitting restart gradio in the settings pane
| -rw-r--r-- | modules/ui.py | 4 |
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=[],
|