1 files changed, 1 insertions, 1 deletions

diff --git a/train_lora.py b/train_lora.py
index 0a3d4c9..e213e3d 100644
--- a/train_lora.py
+++ b/train_lora.py
@@ -398,7 +398,7 @@ def main():
 
     accelerator = Accelerator(
         log_with=LoggerType.TENSORBOARD,
-        logging_dir=f"{output_dir}",
+        project_dir=f"{output_dir}",
         mixed_precision=args.mixed_precision
     )