ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-01-26 16:39:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

bug fix

Browse Source
This commit is contained in:
Jaret Burkett
2023-09-10 09:52:14 -06:00
parent 34bfeba229
commit a437aed45f
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
toolkit/dataloader_mixins.py
View File

@@ -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