run.sh 533 Bytes
Newer Older
luopl's avatar
luopl committed
1
2
3
4
5
6
7
8
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 "一名宇航员在月球上发现一块石碑,上面印有“stepfun”字样,闪闪发光" --height 200 --width 200 --infer_steps 20  --cfg_scale 9.0 --time_shift 13.0