From 0640cdf56909ef4b148b63644619626cd0603fb8 Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Sun, 20 Oct 2024 07:04:19 -0600 Subject: [PATCH] Handle errors in loading size database --- toolkit/data_loader.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/toolkit/data_loader.py b/toolkit/data_loader.py index 1a842dd2..103125e7 100644 --- a/toolkit/data_loader.py +++ b/toolkit/data_loader.py @@ -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 = {}