Parcourir la source

use new shared.args.lora_dir

Alex "mcmonkey" Goodwin il y a 2 ans
Parent
commit
ec6224f556
1 fichiers modifiés avec 1 ajouts et 2 suppressions
  1. 1 2
      modules/training.py

+ 1 - 2
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 ==
     # == Input validation / processing ==
     yield "Prepping..."
     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:
     if dataset is None:
         return "**Missing dataset choice input, cannot continue.**"
         return "**Missing dataset choice input, cannot continue.**"
     if format is None:
     if format is None: