From ec6224f5561ce200ef8c98f967c3f6edafd2ffb0 Mon Sep 17 00:00:00 2001
From: "Alex \"mcmonkey\" Goodwin" <git_commits@alexgoodwin.dev>
Date: Mon, 27 Mar 2023 20:04:16 -0700
Subject: [PATCH] use new shared.args.lora_dir

---
 modules/training.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/modules/training.py b/modules/training.py
index 52ecc55e..0d54a251 100644
--- a/modules/training.py
+++ b/modules/training.py
@@ -92,8 +92,7 @@ def do_train(lora_name: str, micro_batch_size: int, batch_size: int, epochs: int
 
     # == Input validation / processing ==
     yield "Prepping..."
-    # TODO: --lora-dir PR once pulled will need to be applied here
-    lora_name = f"loras/{clean_path(None, lora_name)}"
+    lora_name = f"{shared.args.lora_dir}/{clean_path(None, lora_name)}"
     if dataset is None:
         return "**Missing dataset choice input, cannot continue.**"
     if format is None: