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

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_size_file = os.path.join(dataset_folder, '.aitk_size.json')
if os.path.exists(dataset_size_file):
with open(dataset_size_file, 'r') as f:
self.size_database = json.load(f)
try:
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:
self.size_database = {}