diff --git a/library/train_util.py b/library/train_util.py index c5a605d93..a0496d3dd 100644 --- 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(),