ImageFolder mode of the Image

🤗Datasets
1

Hello

I have seen there is an option to load image with ‘mode’ = ‘RGB’, but I could not see a proper way to pass this argument while creating the Image feature for the dataset.

dataset = load_dataset(
            'imagefolder',
            data_dir=cfg.data.dataset_name_or_path,
            mode='rgb'
)
## causes error....

How can I deal with mixed folder containing RGB / RGBA? I would like all the folders would have the same format

Thanks!