ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-03-02 09:09:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

Change flux latent spact if so it will not use old cache

Browse Source
This commit is contained in:
Jaret Burkett
2024-08-14 11:27:40 -06:00
parent 7fed4ea761
commit 9001e5c933
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

@@ -1362,7 +1362,7 @@ class LatentCachingMixin:
elif self.sd.is_auraflow:
file_item.latent_space_version = 'sdxl'
elif self.sd.is_flux:
file_item.latent_space_version = 'flux'
file_item.latent_space_version = 'flux1'
elif self.sd.model_config.is_pixart_sigma:
file_item.latent_space_version = 'sdxl'
else: