Merge pull request #1215 from bghira/bugfix/sana-vae-cache

vaecache: sana should grab .latent object
bghira · Dec 13, 2024 · b323d1b · b323d1b
2 parents 749ae60 + 4042365
commit b323d1b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/helpers/caching/vae.py b/helpers/caching/vae.py
@@ -532,7 +532,9 @@ def encode_images(self, images, filepaths, load_from_cache=True):
                         latents_uncached - self.vae.config.shift_factor
                     ) * self.vae.config.scaling_factor
                 else:
-                    latents_uncached = latents_uncached * self.vae.config.scaling_factor
+                    latents_uncached = (
+                        latents_uncached.latent * self.vae.config.scaling_factor
+                    )
                 logger.debug(f"Latents shape: {latents_uncached.shape}")
 
             # Prepare final latents list by combining cached and newly computed latents