From 5d88351bb589b64a70194e04d9637ca274135cf4 Mon Sep 17 00:00:00 2001
From: Isotr0py <2037008807@qq.com>
Date: Fri, 25 Aug 2023 11:07:02 +0800
Subject: [PATCH] support jpeg xl

---
 library/train_util.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/library/train_util.py b/library/train_util.py
index c5a605d9..a0496d3d 100644
--- a/library/train_util.py
+++ b/library/train_util.py
@@ -96,6 +96,13 @@ try:
 except:
     pass
 
+try:
+    from jxlpy import JXLImagePlugin
+
+    IMAGE_EXTENSIONS.extend([".jxl", ".JXL"])
+except:
+    pass
+
 IMAGE_TRANSFORMS = transforms.Compose(
     [
         transforms.ToTensor(),