bug with PivotalTuningDatasetCapation

[psy2013GitHub]

the class define: https://github.com/cloneofsimo/lora/blob/master/lora_diffusion/dataset.py#L119 bug: the self.instance_images_path did not align with self.captions

blow is details

first both align well self.instance_images_path = list(set(possibily_src_images)) self.captions = [ x.split("/")[-1].split(".")[0] for x in self.instance_images_path ]

then here is the bug, self.instance_images_path is sorted but not self.captions self.instance_images_path = sorted(self.instance_images_path)