allow flipping for point of interesting autocropping. allow num repeats. Fixed some bugs with new free u

This commit is contained in:
Jaret Burkett
2023-10-12 21:02:47 -06:00
parent 4e3b2c2569
commit 38e441a29c
5 changed files with 27 additions and 4 deletions

View File

@@ -377,6 +377,10 @@ class AiToolkitDataset(LatentCachingMixin, BucketsMixin, CaptionMixin, Dataset):
# keys are file paths
file_list = list(self.caption_dict.keys())
if self.dataset_config.num_repeats > 1:
# repeat the list
file_list = file_list * self.dataset_config.num_repeats
# this might take a while
print(f" - Preprocessing image dimensions")
bad_count = 0