feat: add run_with_accelerate.sh helper script
This commit is contained in:
10
training/scripts/run_with_accelerate.sh
Executable file
10
training/scripts/run_with_accelerate.sh
Executable file
@@ -0,0 +1,10 @@
|
||||
#!/bin/bash
|
||||
# Run training with distributed setup
|
||||
|
||||
set -e
|
||||
|
||||
# Install accelerate if not present
|
||||
pip install -q accelerate
|
||||
|
||||
# Launch with accelerate
|
||||
accelerate launch train-on-this-server.sh --config training/configs/llama2-7b-lora.yaml
|
||||
Reference in New Issue
Block a user