#!/bin/bash
# You better have two terminal, one for the remote server, and one for DiT
CUDA_VISIBLE_DEVICES=7 python fastvideo/sample/call_remote_server_stepvideo.py --model_dir data/stepvideo-t2v/ &

parallel=4
url='127.0.0.1'
model_dir=data/stepvideo-t2v
torchrun --nproc_per_node $parallel fastvideo/sample/sample_t2v_stepvideo.py \
    --model_dir $model_dir \
    --vae_url $url \
    --caption_url $url  \
    --prompt assets/prompt.txt \
    --infer_steps 50  \
    --width 768 \
    --height 768 \
    --num_frames 204 \
    --cfg_scale 9.0 \
    --save_path outputs/ \
    --time_shift 13.0