Fix final LoRA checkpoint report

This commit is contained in:
Xin Wang
2026-08-25 09:45:47 +08:00
parent aa4278e13e
commit 56c8fe0a0e

View File

@@ -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: "