Update scripts/finetune.py

Co-authored-by: NanoCode012 <kevinvong@rocketmail.com>
2023-05-21 23:00:28 -04:00
parent ae1719d30c
commit 3457810988
1 changed files with 1 additions and 1 deletions
--- 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