VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-09 06:45:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

Support JPEG-XL on windows

Browse Source
This commit is contained in:
Isotr0py
2023-10-04 21:48:25 +08:00
committed by GitHub
parent 2d87bb648f
commit 592014923f
1 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
library/train_util.py
View File

@@ -96,6 +96,7 @@ try:
except: except:
pass pass
# JPEG-XL on Linux
try: try:
from jxlpy import JXLImagePlugin from jxlpy import JXLImagePlugin
@@ -103,6 +104,14 @@ try:
except: except:
pass pass
# JPEG-XL on Windows
try:
import pillow_jxl
IMAGE_EXTENSIONS.extend([".jxl", ".JXL"])
except:
pass
IMAGE_TRANSFORMS = transforms.Compose( IMAGE_TRANSFORMS = transforms.Compose(
[ [
transforms.ToTensor(), transforms.ToTensor(),