hiyouga 50c71dd29f update example docs
Former-commit-id: 102cd42768d9eb2cf1219309a25b41e26149067e
2024-05-06 22:51:02 +08:00

6 lines
170 B
Bash

#!/bin/bash
# ZeRO-3 enables weight sharding on multiple GPUs
deepspeed --include "localhost:0,1,2,3" \
src/train.py examples/lora_multi_gpu/llama3_lora_sft_ds.yaml