parallel=2  # or parallel=8
url='127.0.0.1'
model_dir=/home/luopl1/stepfun-ai/stepvideo-t2v

tp_degree=2
ulysses_degree=1
# make sure tp_degree x ulysses_degree = parallel
torchrun --nproc_per_node $parallel run_parallel.py --model_dir $model_dir --vae_url $url --caption_url $url  --ulysses_degree $ulysses_degree --tensor_parallel_degree $tp_degree --prompt "一群女孩在跳芭蕾舞" --height 200 --width 200 --infer_steps 20  --cfg_scale 9.0 --time_shift 13.0