dataset as a folder of images. The training script uses the ImageFolder builder from 🤗 Datasets to automatically build a dataset from the folder. Your directory structure should look like:   Copied data_dir/xxx.png
data_dir/xxy.png
data_dir/[...]/xxz.png Pass the path to the dataset directory to the --train_data_dir argument, and then you can start training:   Copied accelerate launch train_unconditional.py \
    --train_data_dir <path-to-train