From 78e75fd040e8777effdcf293fb52933bd76628de Mon Sep 17 00:00:00 2001 From: Manuel Schmid Date: Sun, 18 Feb 2024 23:09:39 +0100 Subject: [PATCH] fix: use base_model and not fallback key as value for default_model now correctly loads fallback models --- webui.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/webui.py b/webui.py index 75b6104e..e444da86 100644 --- a/webui.py +++ b/webui.py @@ -638,7 +638,7 @@ with shared.gradio_root: embeddings_downloads = preset_prepared['embeddings_downloads'] lora_downloads = preset_prepared['lora_downloads'] - preset_prepared['Base Model'], preset_prepared['lora_downloads'] = launch.download_models( + preset_prepared['base_model'], preset_prepared['lora_downloads'] = launch.download_models( default_model, previous_default_models, checkpoint_downloads, embeddings_downloads, lora_downloads) return modules.meta_parser.load_parameter_button_click(json.dumps(preset_prepared), is_generating)