diff --git a/finetuning/sft_12hz.py b/finetuning/sft_12hz.py
index c1f3f468..4364556b 100644
--- a/finetuning/sft_12hz.py
+++ b/finetuning/sft_12hz.py
@@ -39,9 +39,23 @@ def train():
     parser.add_argument("--lr", type=float, default=2e-5)
     parser.add_argument("--num_epochs", type=int, default=3)
     parser.add_argument("--speaker_name", type=str, default="speaker_test")
+    parser.add_argument(
+        "--logging_dir",
+        type=str,
+        default=None,
+        help="Directory used for Accelerate TensorBoard logs. Defaults to <output_model_path>/logs.",
+    )
     args = parser.parse_args()
 
-    accelerator = Accelerator(gradient_accumulation_steps=4, mixed_precision="bf16", log_with="tensorboard")
+    logging_dir = args.logging_dir or os.path.join(args.output_model_path, "logs")
+    os.makedirs(logging_dir, exist_ok=True)
+
+    accelerator = Accelerator(
+        gradient_accumulation_steps=4,
+        mixed_precision="bf16",
+        log_with="tensorboard",
+        project_dir=logging_dir,
+    )
 
     MODEL_PATH = args.init_model_path