diff options
| author | space-nuko <24979496+space-nuko@users.noreply.github.com> | 2023-03-11 00:42:46 +0000 |
|---|---|---|
| committer | space-nuko <24979496+space-nuko@users.noreply.github.com> | 2023-03-11 00:42:46 +0000 |
| commit | ac38ad7e60bb0ff3194536a72dd1259edad0b30a (patch) | |
| tree | 7bd8f6611874a833298c77dc8528f569338f373e /modules/sd_models.py | |
| parent | 716a69237cefb385f71105dbbf50e92d664e0f42 (diff) | |
| parent | 0cc0ee1bcb4c24a8c9715f66cede06601bfc00c8 (diff) | |
| download | stable-diffusion-webui-gfx803-ac38ad7e60bb0ff3194536a72dd1259edad0b30a.tar.gz stable-diffusion-webui-gfx803-ac38ad7e60bb0ff3194536a72dd1259edad0b30a.tar.bz2 stable-diffusion-webui-gfx803-ac38ad7e60bb0ff3194536a72dd1259edad0b30a.zip | |
Merge remote-tracking branch 'origin/master' into unipc
Diffstat (limited to 'modules/sd_models.py')
| -rw-r--r-- | modules/sd_models.py | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/modules/sd_models.py b/modules/sd_models.py index d847d358..93959f55 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -105,9 +105,15 @@ def checkpoint_tiles(): def list_models():
checkpoints_list.clear()
checkpoint_alisases.clear()
- model_list = modelloader.load_models(model_path=model_path, command_path=shared.cmd_opts.ckpt_dir, ext_filter=[".ckpt", ".safetensors"], ext_blacklist=[".vae.safetensors"])
cmd_ckpt = shared.cmd_opts.ckpt
+ if shared.cmd_opts.no_download_sd_model or cmd_ckpt != shared.sd_model_file or os.path.exists(cmd_ckpt):
+ model_url = None
+ else:
+ model_url = "https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/v1-5-pruned-emaonly.safetensors"
+
+ model_list = modelloader.load_models(model_path=model_path, model_url=model_url, command_path=shared.cmd_opts.ckpt_dir, ext_filter=[".ckpt", ".safetensors"], download_name="v1-5-pruned-emaonly.safetensors", ext_blacklist=[".vae.ckpt", ".vae.safetensors"])
+
if os.path.exists(cmd_ckpt):
checkpoint_info = CheckpointInfo(cmd_ckpt)
checkpoint_info.register()
|