diff --git a/training/configs/ornith-35b-lora.yaml b/training/configs/ornith-35b-lora.yaml
index ba09583..fedab50 100644
--- a/training/configs/ornith-35b-lora.yaml
+++ b/training/configs/ornith-35b-lora.yaml
@@ -41,7 +41,7 @@ train_params:
   logging_steps: 10
   save_steps: 100
   save_total_limit: 3
-  output_dir: ../../output/llama2-7b-lora
+  output_dir: ../../output/ornith-35b-lora
 
 # Precision
 mixed_precision: bf16