fix: use correct path for training script

2026-07-01 16:21:47 -04:00
parent 5eae235180
commit ff0d6e8736
1 changed files with 1 additions and 1 deletions
--- 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!"