Update src/axolotl/utils/models.py for info msg

Co-authored-by: NanoCode012 <kevinvong@rocketmail.com>
2023-05-21 23:01:35 -04:00
parent 3457810988
commit 1b3e401241
1 changed files with 1 additions and 0 deletions
--- a/src/axolotl/utils/models.py
+++ b/src/axolotl/utils/models.py
@@ -293,6 +293,7 @@ def load_llama_adapter(model, cfg):
    )

    if cfg.lora_model_dir:
+        logging.info("Loading pretained LORA")
        model = PeftModel.from_pretrained(
            model,
            cfg.lora_model_dir,