hiyouga 92cafef325 update example docs
Former-commit-id: f02f87c6fbd20adae105c83526baa23dba2042fd
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