Restored functional trainer

1 files changed, 0 insertions, 8 deletions

diff --git a/data/csv.py b/data/csv.py
index 5de3ac7..2a8115b 100644
--- a/data/csv.py
+++ b/data/csv.py
@@ -15,9 +15,6 @@ from data.keywords import prompt_to_keywords, keywords_to_prompt
 from models.clip.util import unify_input_ids
 
 
-image_cache: dict[str, Image.Image] = {}
-
-
 interpolations = {
     "linear": transforms.InterpolationMode.NEAREST,
     "bilinear": transforms.InterpolationMode.BILINEAR,
@@ -27,14 +24,9 @@ interpolations = {
 
 
 def get_image(path):
-    if path in image_cache:
-        return image_cache[path]
-
     image = Image.open(path)
     if not image.mode == "RGB":
         image = image.convert("RGB")
-    image_cache[path] = image
-
     return image