From 1d4c578ac8aaab23cfc5e487e4bbef6cb0e95614 Mon Sep 17 00:00:00 2001 From: JM Date: Sun, 28 Sep 2025 14:04:11 +0300 Subject: [PATCH] move color profile creation inside try --- library/train_util.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/library/train_util.py b/library/train_util.py index deb6dbc4..0d294e7f 100644 --- a/library/train_util.py +++ b/library/train_util.py @@ -2496,9 +2496,9 @@ def load_image(image_path, alpha=False): # Convert image to sRGB icc = image.info.get('icc_profile', '') if icc: - src_profile = ImageCms.ImageCmsProfile( BytesIO(icc) ) - srgb_profile = ImageCms.createProfile("sRGB") try: + src_profile = ImageCms.ImageCmsProfile( BytesIO(icc) ) + srgb_profile = ImageCms.createProfile("sRGB") ImageCms.profileToProfile(image, src_profile, srgb_profile, inPlace=True) image.info["icc_profile"] = ImageCms.ImageCmsProfile(srgb_profile).tobytes() except Exception as e: