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

remove debug print

Browse Source
This commit is contained in:
kohya-ss
2024-09-26 22:22:06 +09:00
parent 9249d00311
commit 24b1fdb664
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
library/train_util.py
View File

@@ -1031,10 +1031,10 @@ class BaseDataset(torch.utils.data.Dataset):
# if the modulo of num_processes is not equal to process_index, skip caching
# this makes each process cache different latents
if i % num_processes != process_index:
if i % num_processes != process_index:
continue
print(f"{process_index}/{num_processes} {i}/{len(image_infos)} {info.latents_npz}")
# print(f"{process_index}/{num_processes} {i}/{len(image_infos)} {info.latents_npz}")
cache_available = caching_strategy.is_disk_cached_latents_expected(
info.bucket_reso, info.latents_npz, subset.flip_aug, subset.alpha_mask