Update sd-webui-hires-i2i/scripts/hires_fix_i2i.py

sd-webui-hires-i2i/scripts/hires_fix_i2i.py

@@ -38,9 +38,8 @@ class HiresUnified(scripts.Script):
 
     def __init__(self):
         super().__init__()
-        # Кэш апскейла init-изображения (режим Before) — чтобы не апскейлить повторно
-        self._before_cache_key: Optional[Tuple[int, str, float]] = None
-        self._before_cache_img: Optional[Image.Image] = None
+        # Кэш апскейла init-изображений (режим Before) — dict для поддержки множественных
+        self._before_cache: dict[Tuple[int, str, float], Image.Image] = {}
 
         # Отложенный апскейл (режим After / txt2img) — параметры сохраняем в setup()
         self._defer_after: bool = False
@@ -187,14 +186,18 @@ class HiresUnified(scripts.Script):
 
         # Если есть init_images И выбран режим "Before" → апскейлим init ДО генерации
         if getattr(p, "init_images", None) and p.init_images and not process_after:
-            init_image = p.init_images[0]
-            key = (self._get_image_bytes_key(init_image), upscaler_name, float(ratio))
-            if key != self._before_cache_key:
-                up = self._resize_with_upscaler(init_image, ratio, upscaler_name)
-                self._before_cache_key = key
-                self._before_cache_img = up
-            if self._before_cache_img is not None:
-                p.init_images[0] = self._before_cache_img
+            first_w, first_h = None, None
+            for i, init_image in enumerate(p.init_images):
+                key = (self._get_image_bytes_key(init_image), upscaler_name, float(ratio))
+                if key not in self._before_cache:
+                    up = self._resize_with_upscaler(init_image, ratio, upscaler_name)
+                    self._before_cache[key] = up
+                p.init_images[i] = self._before_cache[key]
+                if i == 0:
+                    first_w, first_h = p.init_images[i].width, p.init_images[i].height
+            if first_w is not None:
+                p.width = first_w
+                p.height = first_h
             # Ничего не откладываем на postprocess
             self._defer_after = False
         else:
@@ -245,7 +248,9 @@ class HiresUnified(scripts.Script):
                 new_images.append(self._resize_with_upscaler(img, r, name))
             if new_images:
                 processed.images = new_images
+                processed.image = new_images[0]
+                processed.grid = new_images[0]
                 if getattr(processed, "extra_generation_params", None) is not None:
                     processed.extra_generation_params.update({
                         "Hires Applied After": True
-                    })
+                    })