From 962bd2fb7cc15b98adc754072e933f15a05c73d6 Mon Sep 17 00:00:00 2001 From: Qinwen Xu Date: Mon, 22 Apr 2024 23:18:03 +0000 Subject: [PATCH] add back eval compilation cache --- src/maxdiffusion/eval.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/maxdiffusion/eval.py b/src/maxdiffusion/eval.py index 3c052892..ed918c8c 100644 --- a/src/maxdiffusion/eval.py +++ b/src/maxdiffusion/eval.py @@ -38,6 +38,7 @@ from keras.preprocessing.image import ImageDataGenerator from tqdm import tqdm from PIL import Image +cc.initialize_cache(os.path.expanduser("~/jax_cache")) def load_captions(file_path): captions_df = pd.read_csv(file_path, delimiter='\t', header=0, names=['image_id','id', 'caption'])