Add extended log msg, fix lora model ui paths

d8ahazard · Jan 30, 2023 · 49bcd84 · 49bcd84
1 parent bf25ff5
commit 49bcd84
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/dreambooth/train_dreambooth.py b/dreambooth/train_dreambooth.py
@@ -502,6 +502,7 @@ def collate_fn(examples):
         print(f"  UNET: {args.train_unet}")
         print(f"  Freeze CLIP Normalization Layers: {args.freeze_clip_normalization}")
         print(f"  LR: {args.learning_rate}")
+        if args.use_lora_extended: print(f"  LoRA Extended: {args.use_lora_extended}")
         if args.use_lora and stop_text_percentage > 0: print(f"  LoRA Text Encoder LR: {args.lora_txt_learning_rate}")
         print(f"  V2: {args.v2}")
 

diff --git a/dreambooth/utils/model_utils.py b/dreambooth/utils/model_utils.py
@@ -26,7 +26,7 @@ def get_db_models():
 
 def get_lora_models():
     model_dir = shared.lora_models_path
-    out_dir = os.path.join(model_dir, "lora")
+    out_dir = model_dir
     output = [""]
     if os.path.exists(out_dir):
         dirs = os.listdir(out_dir)