From fd338e67bbca13deff18878e4308e5724784bd78 Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Sat, 26 Aug 2023 18:07:24 -0600 Subject: [PATCH] Fixed bug with dataloader not seperating mulitple datasets --- toolkit/data_loader.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/toolkit/data_loader.py b/toolkit/data_loader.py index 5393aa8a..cc50c0d9 100644 --- a/toolkit/data_loader.py +++ b/toolkit/data_loader.py @@ -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 = [