support jpeg xl

kohya-ss · Aug 25, 2023 · 5d88351 · 5d88351
1 parent a46a478
commit 5d88351
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/library/train_util.py b/library/train_util.py
@@ -96,6 +96,13 @@
 except:
     pass
 
+try:
+    from jxlpy import JXLImagePlugin
+
+    IMAGE_EXTENSIONS.extend([".jxl", ".JXL"])
+except:
+    pass
+
 IMAGE_TRANSFORMS = transforms.Compose(
     [
         transforms.ToTensor(),