fix upscale loop erroneously applied multiple times

AUTOMATIC1111 · Nov 4, 2022 · 30b1bcc · 30b1bcc
1 parent 822210b
commit 30b1bcc
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/modules/upscaler.py b/modules/upscaler.py
@@ -57,10 +57,18 @@ def upscale(self, img: PIL.Image, scale: int, selected_model: str = None):
         self.scale = scale
         dest_w = img.width * scale
         dest_h = img.height * scale
+
         for i in range(3):
-            if img.width > dest_w and img.height > dest_h:
-                break
+            shape = (img.width, img.height)
+
             img = self.do_upscale(img, selected_model)
+
+            if shape == (img.width, img.height):
+                break
+
+            if img.width >= dest_w and img.height >= dest_h:
+                break
+
         if img.width != dest_w or img.height != dest_h:
             img = img.resize((int(dest_w), int(dest_h)), resample=LANCZOS)