From 8ce613bb3af2dcdbabe460a0665f80757ffd9796 Mon Sep 17 00:00:00 2001
From: w-e-w <40751091+w-e-w@users.noreply.github.com>
Date: Sat, 19 Aug 2023 16:50:43 +0900
Subject: [PATCH] switch to PNG when images too large

---
 modules/images.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/modules/images.py b/modules/images.py
index a6b4fb1e6..eb6447338 100644
--- a/modules/images.py
+++ b/modules/images.py
@@ -600,6 +600,11 @@ def save_image(image, path, basename, seed=None, prompt=None, extension='png', i
     """
     namegen = FilenameGenerator(p, seed, prompt, image)
 
+    # WebP and JPG formats have maximum dimension limits of 16383 and 65535 respectively. switch to PNG which has a much higher limit
+    if (image.height > 65535 or image.width > 65535) and extension.lower() in ("jpg", "jpeg") or (image.height > 16383 or image.width > 16383) and extension.lower() == "webp":
+        print('Image dimensions too large; saving as PNG')
+        extension = ".png"
+
     if save_to_dirs is None:
         save_to_dirs = (grid and opts.grid_save_to_dirs) or (not grid and opts.save_to_dirs and not no_prompt)