torchrun --nproc_per_node=8 \
    --master_port=29501 \
    causvid/train_ode.py \
    --config_path configs/wan_causal_ode.yaml