From ff0d6e8736a659488f21fc8e54ab50cca0cd8569 Mon Sep 17 00:00:00 2001 From: Christian Medina <37550954+cmedinasoriano@users.noreply.github.com> Date: Wed, 1 Jul 2026 16:21:47 -0400 Subject: [PATCH] fix: use correct path for training script --- train-on-this-server.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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