fix: use correct path for training script
This commit is contained in:
@@ -11,6 +11,6 @@ pip install -q transformers accelerate datasets trl peft bitsandbytes deepspeed
|
|||||||
|
|
||||||
# Run training
|
# Run training
|
||||||
echo "Starting training with accelerate..."
|
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!"
|
echo "Training completed!"
|
||||||
Reference in New Issue
Block a user