From ae1719d30c8dc777460b50ffddb15adf2a376887 Mon Sep 17 00:00:00 2001
From: Wing Lian <wing.lian@gmail.com>
Date: Sun, 21 May 2023 23:00:23 -0400
Subject: [PATCH] Update scripts/finetune.py for logging

Co-authored-by: NanoCode012 <kevinvong@rocketmail.com>
---
 scripts/finetune.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/scripts/finetune.py b/scripts/finetune.py
index 81d29ad28..5c50aeb85 100644
--- 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: