ColossalAI/examples/tutorial/sequence_parallel/test_ci.sh

8 lines
119 B
Bash

#!/bin/bash
set -euxo pipefail
pip install -r requirements.txt
# run test
colossalai run --nproc_per_node 4 train.py