github/ai-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions 1 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): class AiToolkitDataset(Dataset, CaptionMixin, BucketsMixin):
file_list: List['FileItem'] = []
def __init__(self, dataset_config: 'DatasetConfig', batch_size=1): def __init__(self, dataset_config: 'DatasetConfig', batch_size=1):
super().__init__() super().__init__()
@@ -326,6 +325,7 @@ class AiToolkitDataset(Dataset, CaptionMixin, BucketsMixin):
# we always random crop if random scale is enabled # 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.random_crop = self.random_scale if self.random_scale else dataset_config.random_crop
self.resolution = dataset_config.resolution self.resolution = dataset_config.resolution
self.file_list: List['FileItem'] = []
# get the file list # get the file list
file_list = [ file_list = [