Merge pull request #856 from Isotr0py/jxl

Fix JPEG-XL support
kohya-ss · Oct 9, 2023 · 5fe52ed · 5fe52ed
2 parents 8b247a3 + 5920149
commit 5fe52ed
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/library/train_util.py b/library/train_util.py
@@ -96,13 +96,22 @@
 except:
     pass
 
+# JPEG-XL on Linux
 try:
     from jxlpy import JXLImagePlugin
 
     IMAGE_EXTENSIONS.extend([".jxl", ".JXL"])
 except:
     pass
 
+# JPEG-XL on Windows
+try:
+    import pillow_jxl
+
+    IMAGE_EXTENSIONS.extend([".jxl", ".JXL"])
+except:
+    pass
+
 IMAGE_TRANSFORMS = transforms.Compose(
     [
         transforms.ToTensor(),