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

Fixed bug with dataloader not seperating mulitple datasets

Browse Source
This commit is contained in:
Jaret Burkett
2023-08-26 18:07:24 -06:00
parent 8105c05c12
commit fd338e67bb
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
toolkit/data_loader.py
View File

@@ -312,7 +312,6 @@ class FileItem:
class AiToolkitDataset(Dataset, CaptionMixin, BucketsMixin):
file_list: List['FileItem'] = []
def __init__(self, dataset_config: 'DatasetConfig', batch_size=1):
super().__init__()
@@ -326,6 +325,7 @@ class AiToolkitDataset(Dataset, CaptionMixin, BucketsMixin):
# we always random crop if random scale is enabled
self.random_crop = self.random_scale if self.random_scale else dataset_config.random_crop
self.resolution = dataset_config.resolution
self.file_list: List['FileItem'] = []
# get the file list
file_list = [