VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-09 06:45:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

show index in caching latents

Browse Source
This commit is contained in:
Kohya S
2023-03-02 21:32:02 +09:00
parent 09f575fd4d
commit 2d2407410e
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
library/train_util.py
View File

@@ -1061,7 +1061,8 @@ class DatasetGroup(torch.utils.data.ConcatDataset):
# dataset.make_buckets() # dataset.make_buckets()
def cache_latents(self, vae): def cache_latents(self, vae):
for dataset in self.datasets: for i, dataset in enumerate(self.datasets):
print(f"[Dataset {i}]")
dataset.cache_latents(vae) dataset.cache_latents(vae)
def is_latent_cacheable(self) -> bool: def is_latent_cacheable(self) -> bool: