 
export frame_num=65

 python wan/bench_speed/generate_i2v_batch.py --dataset_path wan/bench_speed/<dataset_path> \
                    --output_dir speed_testi2v --sample_step 20 \
                    --frame_num $frame_num \
                    --ckpt_dir  ./Wan2.2-TI2V-5B  --task ti2v-5B  --size "640*736"


python wan/bench_speed/generate_causal_batch_server.py \
        --task ti2v-5B \
        --size "640*736" \
        --ckpt_dir ./Wan2.2-TI2V-5B \
        --convert_model_dtype \
        --sample_steps 5 \
        --frame_num $frame_num \
        --dataset_path wan/bench_speed/<dataset_path>\
        --output_dir speed_test_causalserver


python wan/bench_speed/generate_causal_batch_server_prefill_all.py \
        --task ti2v-5B \
        --size "640*736" \
        --ckpt_dir ./Wan2.2-TI2V-5B \
        --convert_model_dtype \
        --sample_steps 5 \
        --frame_num $frame_num \
        --dataset_path wan/bench_speed/<dataset_path>\
        --output_dir speed_test_causalserver