# export GPU_FLUSH_ON_EXECUTION=1

len=33
step=20

for num in  1 2 4;do
    torchrun --nproc_per_node=${num} sample_video.py \
        --video-size 1280 720 \
        --video-length ${len} \
        --infer-steps ${step} \
        --prompt "A cat walks on the grass, realistic style." \
        --flow-reverse \
        --seed 42 \
        --ulysses-degree ${num} \
        --ring-degree 1 \
        --save-path ./results 2>&1 |tee video-logs/bw-video-len_${len}-step_${step}-num-${num}.log
done