From 964fc13a99d47263d023f4e3116ac2c220acec88 Mon Sep 17 00:00:00 2001
From: w-e-w <40751091+w-e-w@users.noreply.github.com>
Date: Sat, 20 Jul 2024 04:00:54 +0900
Subject: [PATCH] fix upscale logic

---
 modules/upscaler.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/modules/upscaler.py b/modules/upscaler.py
index 28c60cdcdfe..507881fede2 100644
--- a/modules/upscaler.py
+++ b/modules/upscaler.py
@@ -56,8 +56,8 @@ def upscale(self, img: PIL.Image, scale, selected_model: str = None):
         dest_w = int((img.width * scale) // 8 * 8)
         dest_h = int((img.height * scale) // 8 * 8)
 
-        for _ in range(3):
-            if img.width >= dest_w and img.height >= dest_h and scale != 1:
+        for i in range(3):
+            if img.width >= dest_w and img.height >= dest_h and (i > 0 or scale != 1):
                 break
 
             if shared.state.interrupted: