run.sh 457 Bytes
Newer Older
luopl's avatar
luopl committed
1
2
3
4
5
parallel=1  # or parallel=8 Single GPU can also predict the results, although it will take longer
url='127.0.0.1'
model_dir=/home/luopl1/stepfun-ai/stepvideo-ti2v/

torchrun --nproc_per_node $parallel run_parallel.py --model_dir $model_dir --vae_url $url --caption_url $url  --ulysses_degree  $parallel --prompt "笑起来" --first_image_path ./assets/demo.png --infer_steps 20  --width 400 --height 400 --cfg_scale 9.0 --time_shift 13.0 --motion_score 5.0