diff --git a/train_omni_turn_lora.py b/train_omni_turn_lora.py index c4cff7f..ea8cfc7 100644 --- a/train_omni_turn_lora.py +++ b/train_omni_turn_lora.py @@ -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: "