From 6221ccba4fd418252e2ff2e8df9f9c3bfafdbbe0 Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Fri, 19 May 2023 14:06:46 -0400 Subject: [PATCH] change default model on download --- modules/sd_models.py | 1 + 1 file changed, 1 insertion(+) diff --git a/modules/sd_models.py b/modules/sd_models.py index d637e9190..ef7fea519 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -118,6 +118,7 @@ def list_models(): key = input('Download the default model? (y/N) ') if key.lower().startswith('y'): model_url = "https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/v1-5-pruned-emaonly.safetensors" + shared.opts.data['sd_model_checkpoint'] = "v1-5-pruned-emaonly.safetensors" model_list = modelloader.load_models(model_path=model_path, model_url=model_url, command_path=shared.opts.ckpt_dir, ext_filter=[".ckpt", ".safetensors"], download_name="v1-5-pruned-emaonly.safetensors", ext_blacklist=[".vae.ckpt", ".vae.safetensors"]) for filename in sorted(model_list, key=str.lower): checkpoint_info = CheckpointInfo(filename)