From 620a06f517032fff9842b81950795bb14c0ad361 Mon Sep 17 00:00:00 2001 From: Disty0 Date: Mon, 17 Mar 2025 17:44:29 +0300 Subject: [PATCH] Check for uppercase file extension too --- library/train_util.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/library/train_util.py b/library/train_util.py index 916b8834..3b6f7663 100644 --- a/library/train_util.py +++ b/library/train_util.py @@ -1158,7 +1158,7 @@ class BaseDataset(torch.utils.data.Dataset): ) def get_image_size(self, image_path): - if image_path.endswith(".jxl"): + if image_path.endswith(".jxl") or image_path.endswith(".JXL"): return get_jxl_size(image_path) return imagesize.get(image_path)