python custom/webdataset/prepare_LBM.py \
    --dataset $1    \
    --buffer $2     \
    --num_proc 1    \
    --latents       \
    --delete        \