From 0e6295886d2d0e9309c8223201e8a4bd55e4735c Mon Sep 17 00:00:00 2001 From: oobabooga <112222186+oobabooga@users.noreply.github.com> Date: Wed, 12 Jul 2023 14:19:33 -0700 Subject: [PATCH] Fix lora download folder --- server.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/server.py b/server.py index 05455775..56f94963 100644 --- a/server.py +++ b/server.py @@ -145,8 +145,8 @@ def download_model_wrapper(repo_id, progress=gr.Progress()): links, sha256, is_lora = downloader.get_download_links_from_huggingface(model, branch, text_only=False) yield ("Getting the output folder") - models_dir = Path(shared.args.model_dir) - output_folder = downloader.get_output_folder(model, branch, is_lora, base_folder=models_dir) + base_folder = shared.args.lora_dir if is_lora else shared.args.model_dir + output_folder = downloader.get_output_folder(model, branch, is_lora, base_folder=base_folder) if check: progress(0.5)