tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Update scripts/finetune.py for logging

Browse Source 
Co-authored-by: NanoCode012 <kevinvong@rocketmail.com>
This commit is contained in:
Wing Lian
2023-05-21 23:00:23 -04:00
committed by GitHub
parent 98a6781f18
commit ae1719d30c
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/finetune.py
View File

@@ -171,7 +171,7 @@ def train(
) )
if "merge_lora" in kwargs and cfg.adapter is not None: 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() model = model.merge_and_unload()
if cfg.local_rank == 0: if cfg.local_rank == 0: