From 500d9a32c7b1f877c8f44159a9a10c594b545a80 Mon Sep 17 00:00:00 2001
From: AUTOMATIC <16777216c@gmail.com>
Date: Sat, 21 Jan 2023 23:11:37 +0300
Subject: [PATCH] add --lora-dir commandline option

---
 extensions-builtin/Lora/lora.py                   | 9 ++++-----
 extensions-builtin/Lora/preload.py                | 6 ++++++
 extensions-builtin/Lora/ui_extra_networks_lora.py | 2 +-
 3 files changed, 11 insertions(+), 6 deletions(-)
 create mode 100644 extensions-builtin/Lora/preload.py

diff --git a/extensions-builtin/Lora/lora.py b/extensions-builtin/Lora/lora.py
index 6d860224a..da1797dc9 100644
--- a/extensions-builtin/Lora/lora.py
+++ b/extensions-builtin/Lora/lora.py
@@ -177,12 +177,12 @@ def lora_Conv2d_forward(self, input):
 def list_available_loras():
     available_loras.clear()
 
-    os.makedirs(lora_dir, exist_ok=True)
+    os.makedirs(shared.cmd_opts.lora_dir, exist_ok=True)
 
     candidates = \
-        glob.glob(os.path.join(lora_dir, '**/*.pt'), recursive=True) + \
-        glob.glob(os.path.join(lora_dir, '**/*.safetensors'), recursive=True) + \
-        glob.glob(os.path.join(lora_dir, '**/*.ckpt'), recursive=True)
+        glob.glob(os.path.join(shared.cmd_opts.lora_dir, '**/*.pt'), recursive=True) + \
+        glob.glob(os.path.join(shared.cmd_opts.lora_dir, '**/*.safetensors'), recursive=True) + \
+        glob.glob(os.path.join(shared.cmd_opts.lora_dir, '**/*.ckpt'), recursive=True)
 
     for filename in sorted(candidates):
         if os.path.isdir(filename):
@@ -193,7 +193,6 @@ def list_available_loras():
         available_loras[name] = LoraOnDisk(name, filename)
 
 
-lora_dir = os.path.join(shared.models_path, "Lora")
 available_loras = {}
 loaded_loras = []
 
diff --git a/extensions-builtin/Lora/preload.py b/extensions-builtin/Lora/preload.py
new file mode 100644
index 000000000..863dc5c0b
--- /dev/null
+++ b/extensions-builtin/Lora/preload.py
@@ -0,0 +1,6 @@
+import os
+from modules import paths
+
+
+def preload(parser):
+    parser.add_argument("--lora-dir", type=str, help="Path to directory with Lora networks.", default=os.path.join(paths.models_path, 'Lora'))
diff --git a/extensions-builtin/Lora/ui_extra_networks_lora.py b/extensions-builtin/Lora/ui_extra_networks_lora.py
index 653978901..4406f8a07 100644
--- a/extensions-builtin/Lora/ui_extra_networks_lora.py
+++ b/extensions-builtin/Lora/ui_extra_networks_lora.py
@@ -31,5 +31,5 @@ class ExtraNetworksPageLora(ui_extra_networks.ExtraNetworksPage):
             }
 
     def allowed_directories_for_previews(self):
-        return [lora.lora_dir]
+        return [shared.cmd_opts.lora_dir]