the issue is honestly with the dataset, if I just give it this Yorth/dalleTestDataFiltered2, it will make life much easier than downloading the dataset, separating the image folder, finding a way to link them with the captions without messing things up