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

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 5c50aeb85..cb9d7e94e 100644
--- a/scripts/finetune.py
+++ b/scripts/finetune.py
@@ -175,7 +175,7 @@ def train(
         model = model.merge_and_unload()
 
         if cfg.local_rank == 0:
-            print("saving merged model")
+            logging.info("saving merged model")
             model.save_pretrained(str(Path(cfg.output_dir) / "merged"))
         return