Fix for check with cfg and merge_lora (#600)

2023-09-18 21:14:32 -04:00
parent 31b9e0c6e8
commit 62a774140b
3 changed files with 3 additions and 3 deletions
--- a/src/axolotl/cli/init.py
+++ b/src/axolotl/cli/init.py
@@ -70,7 +70,7 @@ def do_merge_lora(
    model.to(dtype=torch.float16)

    if cfg.local_rank == 0:
-        LOG.info("saving merged model")
+        LOG.info(f"saving merged model to: {str(Path(cfg.output_dir) / 'merged')}")
        model.save_pretrained(
            str(Path(cfg.output_dir) / "merged"),
            safe_serialization=safe_serialization,