DATA_DIR="TODO: replace with your directory"

# Create shards
echo "Creating image shards..."
python -m src.datasets.images.shard_images \
    --create \
    --input_images_dir ${DATA_DIR}/static/raw/wit \
    --output_images_dir ${DATA_DIR}/static/instruct/wit \
    --images_per_shard 512