8ee1677 be76c6b
1
2
3
#!/usr/bin/env bash CUDA_VISIBLE_DEVICES="0,3" torchrun run_model.py --pipeline-model-parallel-size 1 --tensor-model-parallel-size 1