From 3708de2b1f3344940e307b2487ea4a6c0cdaa63d Mon Sep 17 00:00:00 2001
From: micsthepick <11528421+micsthepick@users.noreply.github.com>
Date: Wed, 12 Jul 2023 07:55:46 +1000
Subject: [PATCH] respect model dir for downloads (#3077) (#3079)

---
 server.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/server.py b/server.py
index 18474d3b..412f13ec 100644
--- a/server.py
+++ b/server.py
@@ -145,7 +145,13 @@ 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")
-        output_folder = downloader.get_output_folder(model, branch, is_lora)
+        models_dir = Path(shared.args.model_dir)
+
+        # If the last part of the path is "models", remove it
+        if models_dir.name.lower() == 'models':
+            models_dir = models_dir.parent
+
+        output_folder = downloader.get_output_folder(model, branch, is_lora, base_folder=models_dir)
 
         if check:
             progress(0.5)