Issue with diffusers training notebook when trying to load images from local disk

[GeorvityLabs]

Describe the bug

Reproduction

https://colab.research.google.com/github/huggingface/notebooks/blob/main/diffusers/training_example.ipynb

^notebook i used

Logs

No response

System Info

Ubuntu 20.04

[GeorvityLabs]

any thoughts on how this could be fixed @tmabraham @patil-suraj @pcuenca

[GeorvityLabs]

@anton-l any suggestions on a fix for this

[GeorvityLabs]

^ here I am loading images from local disk , but since the format changes, it's giving this error .
what would be a potential fix ?

[GeorvityLabs]

fixed the issue , in the notebook
change
dataset = load_dataset(config.dataset_name, data_dir="path")
to
dataset = load_dataset(config.dataset_name, data_dir="path",split="train")

[patrickvonplaten]

@anton-l could you check here whether we should update the training colab?