bug fix

2023-09-10 09:52:14 -06:00 · 2023-09-10 09:52:14 -06:00 · a437aed45f
parent 34bfeba229
commit a437aed45f
1 changed files with 1 additions and 1 deletions
--- a/toolkit/dataloader_mixins.py
+++ b/toolkit/dataloader_mixins.py
@ -408,7 +408,7 @@ class LatentCachingMixin:

                if to_memory:
                    # keep it in memory
-                    file_item._encoded_latent = latent.to('cpu', dtype=self.sd.dtype)
+                    file_item._encoded_latent = latent.to('cpu', dtype=self.sd.torch_dtype)

                flush(garbage_collect=False)
            file_item.is_latent_cached = True