Spaces:

Dovakiins
/

qwerrwe

Build error

Nanobit commited on May 22, 2023

Commit

3457810

unverified ·

1 Parent(s): ae1719d

Update scripts/finetune.py

Co-authored-by: NanoCode012 <kevinvong@rocketmail.com>

Files changed (1) hide show

scripts/finetune.py CHANGED Viewed

@@ -175,7 +175,7 @@ def train(
         model = model.merge_and_unload()
         if cfg.local_rank == 0:
-            print("saving merged model")
             model.save_pretrained(str(Path(cfg.output_dir) / "merged"))
         return

         model = model.merge_and_unload()
         if cfg.local_rank == 0:
+            logging.info("saving merged model")
             model.save_pretrained(str(Path(cfg.output_dir) / "merged"))
         return