make sure to set lora_model_dir

axolotl-ai-cloud · Jan 28, 2024 · 27a524c · 27a524c
1 parent b9721c8
commit 27a524c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/axolotl/train.py b/src/axolotl/train.py
@@ -73,7 +73,8 @@ def train(
     resume_from_checkpoint = cfg.resume_from_checkpoint
 
     if cfg.adapter and cfg.resume_from_checkpoint and not cfg.lora_model_dir:
-        LOG.info("")
+        LOG.info(f"setting lora_model_dir to use {cfg.resume_from_checkpoint}")
+        cfg.lora_model_dir = cfg.resume_from_checkpoint
 
     # Load the model and tokenizer
     msg = "loading model"