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