Update scripts/finetune.py for logging

Co-authored-by: NanoCode012 <kevinvong@rocketmail.com>
2023-05-21 23:00:23 -04:00
parent 98a6781f18
commit ae1719d30c
1 changed files with 1 additions and 1 deletions
--- a/scripts/finetune.py
+++ b/scripts/finetune.py
@@ -171,7 +171,7 @@ def train(
    )

    if "merge_lora" in kwargs and cfg.adapter is not None:
-        print("running merge of LoRA with base model")
+        logging.info("running merge of LoRA with base model")
        model = model.merge_and_unload()

        if cfg.local_rank == 0: