File size: 131 Bytes
03d48b6 | 1 2 3 | #!/usr/bin/env bash
CUDA_VISIBLE_DEVICES="0" torchrun run_model.py --pipeline-model-parallel-size 1 --tensor-model-parallel-size 1
|
03d48b6 | 1 2 3 | #!/usr/bin/env bash
CUDA_VISIBLE_DEVICES="0" torchrun run_model.py --pipeline-model-parallel-size 1 --tensor-model-parallel-size 1
|