From 632ce00b54ffeacfc18f44f10827f167ab3ac37c Mon Sep 17 00:00:00 2001 From: Volpeon Date: Sun, 15 Jan 2023 21:06:16 +0100 Subject: Restored functional trainer --- data/csv.py | 8 -------- 1 file changed, 8 deletions(-) (limited to 'data') 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 -- cgit v1.2.3-70-g09d2