Spaces:

Dovakiins
/

qwerrwe

Build error

Nanobit commited on May 22, 2023

Commit

ae1719d

unverified ·

1 Parent(s): 98a6781

Update scripts/finetune.py for logging

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

Files changed (1) hide show

scripts/finetune.py CHANGED Viewed

@@ -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")
         model = model.merge_and_unload()
         if cfg.local_rank == 0:

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