cmedina/agenx-lora-training
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: add --mixed_precision bf16 to accelerate launch

Browse Source
This commit is contained in:
Christian Medina
2026-07-02 09:26:37 -04:00
parent fe7df7c92f
commit 7b39fc3a1b
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
train-on-this-server.sh
View File

@@ -18,9 +18,9 @@ pip install --upgrade pip
echo "Installing training dependencies..." echo "Installing training dependencies..."
pip install transformers datasets trl peft accelerate bitsandbytes deepspeed pip install transformers datasets trl peft accelerate bitsandbytes deepspeed
# Run training with FSDP # Run training with accelerate data parallelism
echo "Starting training with FSDP..." echo "Starting training with accelerate..."
export PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True export PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True
accelerate launch --multi_gpu --num_processes 2 train.py --config training/configs/ornith-35b-lora.yaml accelerate launch --multi_gpu --num_processes 2 --mixed_precision bf16 train.py --config training/configs/ornith-35b-lora.yaml
echo "Training completed!" echo "Training completed!"