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(),