Fix final LoRA checkpoint report
This commit is contained in:
@@ -1616,7 +1616,7 @@ def _run_training(args: argparse.Namespace, run: DistributedRun) -> None:
|
||||
if args.freeze_thinker:
|
||||
print("Frozen Thinker: no LoRA adapter was created")
|
||||
else:
|
||||
print(f"Saved LoRA adapter to {adapter_dir}")
|
||||
print(f"Saved LoRA adapter to {output_dir / 'adapter'}")
|
||||
print(f"Saved turn head to {output_dir / 'turn_head.pt'}")
|
||||
print(
|
||||
"Peak CUDA allocated across ranks: "
|
||||
|
||||
Reference in New Issue
Block a user