github/ai-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Handle errors in loading size database

Browse Source
This commit is contained in:
Jaret Burkett
2024-10-20 07:04:19 -06:00
parent 473d455f44
commit 0640cdf569
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
toolkit/data_loader.py
View File

@@ -442,8 +442,13 @@ class AiToolkitDataset(LatentCachingMixin, CLIPCachingMixin, BucketsMixin, Capti
dataset_folder = os.path.dirname(dataset_folder) dataset_folder = os.path.dirname(dataset_folder)
dataset_size_file = os.path.join(dataset_folder, '.aitk_size.json') dataset_size_file = os.path.join(dataset_folder, '.aitk_size.json')
if os.path.exists(dataset_size_file): if os.path.exists(dataset_size_file):
with open(dataset_size_file, 'r') as f: try:
self.size_database = json.load(f) with open(dataset_size_file, 'r') as f:
self.size_database = json.load(f)
except Exception as e:
print(f"Error loading size database: {dataset_size_file}")
print(e)
self.size_database = {}
else: else:
self.size_database = {} self.size_database = {}