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

Added a file signature check on the dataset size caching system to invalidate cached dimensions if the file changes.

Browse Source
This commit is contained in:
Jaret Burkett
2025-04-01 07:39:36 -06:00
parent 5ea19b6292
commit 3d131fb27a
3 changed files with 25 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
toolkit/data_loader.py
View File

@@ -458,7 +458,7 @@ class AiToolkitDataset(LatentCachingMixin, CLIPCachingMixin, BucketsMixin, Capti
dataset_folder = os.path.dirname(dataset_folder)
dataset_size_file = os.path.join(dataset_folder, '.aitk_size.json')
dataloader_version = "0.1.1"
dataloader_version = "0.1.2"
if os.path.exists(dataset_size_file):
try:
with open(dataset_size_file, 'r') as f: