torchrun --nproc_per_node=4  \
    --nnodes=1 \
    --node_rank=0 \
    --master_port=20023 flashvideo/dist_inf_text_file.py \
    --base "flashvideo/configs/stage1.yaml" \
    --second "flashvideo/configs/stage2.yaml" \
    --inf-ckpt  ./checkpoints/stage1.pt \
    --inf-ckpt2 ./checkpoints/stage2.pt \
    --input-file ./example.txt \
    --output-dir ./vis_270p_1080p_example