From c429bcc5d8ec1710ef3f262a0acd6ec3bea9d474 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Semjon=20Kravt=C5=A1enko?= Date: Tue, 7 Feb 2023 16:32:12 +0200 Subject: [PATCH] Do not convert to JPEG if not possible Do not export for 4chan if conversion to JPEG is not possible. This prevents an exception when trying to save large depthmaps when export_for_4chan setting is on --- modules/images.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/modules/images.py b/modules/images.py index c2ca8849..5f159741 100644 --- a/modules/images.py +++ b/modules/images.py @@ -577,7 +577,8 @@ def save_image(image, path, basename, seed=None, prompt=None, extension='png', i target_side_length = 4000 oversize = image.width > target_side_length or image.height > target_side_length - if opts.export_for_4chan and (oversize or os.stat(fullfn).st_size > 4 * 1024 * 1024): + unsupported_mode = image.mode in ['I', 'F', 'I;16', 'I;16L', 'I;16B', 'I;16N'] + if opts.export_for_4chan and (oversize or os.stat(fullfn).st_size > 4 * 1024 * 1024) and not unsupported_mode: ratio = image.width / image.height if oversize and ratio > 1: