diff --git a/toolkit/dataloader_mixins.py b/toolkit/dataloader_mixins.py index 09a5bf65..536c8498 100644 --- a/toolkit/dataloader_mixins.py +++ b/toolkit/dataloader_mixins.py @@ -234,6 +234,7 @@ class ImageProcessingDTOMixin: # if we are caching latents, just do that if self.is_latent_cached: self.get_latent() + return try: img = Image.open(self.path).convert('RGB') img = exif_transpose(img)