From 257da9493d67736b13cfceb6321900ee020e1130 Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Tue, 12 Sep 2023 18:39:41 -0600 Subject: [PATCH] Dont load image if we are cachine latents --- toolkit/dataloader_mixins.py | 1 + 1 file changed, 1 insertion(+) 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)