diff --git a/train-on-this-server.sh b/train-on-this-server.sh
index 3638aec..8bc478f 100755
--- a/train-on-this-server.sh
+++ b/train-on-this-server.sh
@@ -11,6 +11,6 @@ pip install -q transformers accelerate datasets trl peft bitsandbytes deepspeed
 
 # Run training
 echo "Starting training with accelerate..."
-accelerate launch --multi_gpu --num_processes 2 /home/workdir/artifacts/train.py --config training/configs/llama2-7b-lora.yaml
+accelerate launch --multi_gpu --num_processes 2 training/scripts/train.py --config training/configs/ornith-35b-lora.yaml
 
 echo "Training completed!"
\ No newline at end of file