patrickvonplaten's picture
upload
b9e3eba
raw
history blame contribute delete
131 Bytes
#!/usr/bin/env bash
CUDA_VISIBLE_DEVICES="0" torchrun run_model.py --pipeline-model-parallel-size 1 --tensor-model-parallel-size 1