LLaMA-Factory/examples/full_multi_gpu/predict.sh

6 lines
187 B
Bash
Raw Normal View History

#!/bin/bash
2024-04-23 10:29:46 +00:00
CUDA_VISIBLE_DEVICES=0,1,2,3 accelerate launch \
2024-05-06 14:51:02 +00:00
--config_file examples/accelerate/single_config.yaml \
src/train.py examples/full_multi_gpu/llama3_full_predict.yaml