mrq/stable-diffusion-webui
1
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Do not convert to JPEG if not possible

Browse Source 
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
This commit is contained in:
Semjon Kravtšenko 2023-02-07 16:32:12 +02:00 committed by GitHub
parent ea9bd9fc74
commit c429bcc5d8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
modules/images.py
View File

@ -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: