VAECache: fix for jpg/jpeg images (fix tuple reference, pt 2)

bghira · Dec 23, 2023 · 83eb6e3 · 83eb6e3
1 parent 14bcd49
commit 83eb6e3
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/helpers/caching/vae.py b/helpers/caching/vae.py
@@ -162,9 +162,9 @@ def discover_unprocessed_files(self, directory: str = None):
 
         # Convert cache filenames to their corresponding image filenames
         existing_image_filenames = {
-            os.path.splitext(self._image_filename_from_vaecache_filename(cache_file))[
-                0
-            ][0]
+            os.path.splitext(
+                self._image_filename_from_vaecache_filename(cache_file)[0]
+            )[0]
             for cache_file in existing_cache_files
         }