From a437aed45ff12f58ea7f81ec44abbcab2b3e267e Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Sun, 10 Sep 2023 09:52:14 -0600 Subject: [PATCH] bug fix --- toolkit/dataloader_mixins.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/toolkit/dataloader_mixins.py b/toolkit/dataloader_mixins.py index 2c2ffd59..018c3638 100644 --- 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