Merge pull request #786 from Isotr0py/jxl

Support JPEG XL
kohya-ss · Sep 2, 2023 · 497051c · 497051c
2 parents 6400116 + 5d88351
commit 497051c
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(),