From d8d1e6fd1edec5d51b92fb86c9755fba9e7a3872 Mon Sep 17 00:00:00 2001
From: Jaret Burkett <jaretburkett@gmail.com>
Date: Tue, 12 Sep 2023 18:48:39 -0600
Subject: [PATCH] big fixes

---
 toolkit/data_loader.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/toolkit/data_loader.py b/toolkit/data_loader.py
index 830c8e80..1be02c67 100644
--- a/toolkit/data_loader.py
+++ b/toolkit/data_loader.py
@@ -1,6 +1,7 @@
 import json
 import os
 import random
+from functools import lru_cache
 from typing import List, TYPE_CHECKING
 
 import cv2
@@ -417,6 +418,7 @@ class AiToolkitDataset(LatentCachingMixin, BucketsMixin, CaptionMixin, Dataset):
         file_item.load_caption(self.caption_dict)
         return file_item
 
+    @lru_cache(maxsize=300)
     def __getitem__(self, item):
         if self.dataset_config.buckets:
             # for buckets we collate ourselves for now