jxl image size workaround

feffy380 · May 29, 2024 · a7d2685 · a7d2685
1 parent 5c8e2bd
commit a7d2685
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/library/train_util.py b/library/train_util.py
@@ -1182,7 +1182,10 @@ def cache_text_encoder_outputs(
             )
 
     def get_image_size(self, image_path):
-        return imagesize.get(image_path)
+        if image_path.endswith(".jxl"):
+            return Image.open(image_path).size
+        else:
+            return imagesize.get(image_path)
 
     def load_image_with_face_info(self, subset: BaseSubset, image_path: str):
         img = load_image(image_path)