Upload sd-webui-hires-i2i using SD-Hub

.gitattributes

@@ -142,3 +142,5 @@ DWPose/resources/generation.jpg filter=lfs diff=lfs merge=lfs -text
 DWPose/resources/iron.gif filter=lfs diff=lfs merge=lfs -text
 DWPose/resources/jay_pose.jpg filter=lfs diff=lfs merge=lfs -text
 DWPose/resources/lalaland.gif filter=lfs diff=lfs merge=lfs -text
+sd-webui-hires-i2i/img/off.jpg filter=lfs diff=lfs merge=lfs -text
+sd-webui-hires-i2i/img/on.jpg filter=lfs diff=lfs merge=lfs -text

sd-webui-hires-i2i/.gitignore

@@ -0,0 +1 @@
+__pycache__

sd-webui-hires-i2i/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Haoming
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

sd-webui-hires-i2i/README.md

@@ -0,0 +1,35 @@
+# SD Webui Hires. Fix img2img
+This is an Extension for the [Automatic1111 Webui](https://github.com/AUTOMATIC1111/stable-diffusion-webui), which allows you to upscale your input image with an Upscaler before the **img2img** pipeline.
+
+> Compatible with [Forge](https://github.com/lllyasviel/stable-diffusion-webui-forge)
+
+By default, the Webui uses the `LANCZOS` algorithm to resize when the generation dimension is different from the size of your input image. You can also set the `Upscaler for img2img` option in the Settings to use an Upscaler instead. Alternatively, you can also send the image to the **Extras** tab and upscale first, then send the result back to the **img2img** tab.
+
+**TL;DR:** This Extension just simplifies the processes
+
+<table>
+<thead align="center">
+    <tr>
+        <td><b>Extension</b></td>
+        <td>Off</td>
+        <td>On</td>
+    </tr>
+</thead>
+<tbody align="center">
+    <tr>
+        <td><b>Result</b></td>
+        <td><img src="img/off.jpg"></td>
+        <td><img src="img/on.jpg"></td>
+    </tr>
+</tbody>
+</table>
+
+<details>
+<summary>Infotext</summary>
+
+1. Generate `a photo of a dog` in `512x512`
+2. Resize to `256x256` via **Paint 3D**
+3. Upscale to `1024x1024` via **img2img**, with the Extension disabled and enabled respectively
+    - **Upscaler:** `4xLSDIRplusC`
+
+</details>

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

@@ -0,0 +1,251 @@
+# -*- coding: utf-8 -*-
+"""
+Hires. Fix (txt2img + img2img) — единый скрипт с кнопкой и выбором режима.
+- img2img: апскейл исходного init_image ДО генерации (режим Before).
+- txt2img: апскейл результата ПОСЛЕ генерации (режим After).
+Не падает в txt2img (там нет p.init_images).
+
+Требования/среда:
+- Python 3.10+
+- AUTOMATIC1111 SD WebUI (scripts API: ui, setup, process, postprocess)
+- Gradio 3.x (как в A1111)
+"""
+
+from __future__ import annotations
+
+import re
+from typing import Optional, Tuple
+
+from PIL import Image
+import gradio as gr
+
+from modules import scripts, shared, images
+from modules.ui_components import InputAccordion
+
+
+class HiresUnified(scripts.Script):
+    """
+    Единый скрипт, совместимый с txt2img и img2img.
+
+    Жизненный цикл A1111:
+      - show(is_img2img)             → когда показывать панель
+      - ui(is_img2img)               → построение UI; возвращаем список компонентов
+      - setup(p, *args)              → до запуска генерации; можно менять p
+      - process(p, *args)            → хук до генерации (без processed)
+      - postprocess(p, processed, *args)
+                                     → хук после генерации; тут можно менять processed.images
+    """
+
+    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
+
+        # Отложенный апскейл (режим After / txt2img) — параметры сохраняем в setup()
+        self._defer_after: bool = False
+        self._defer_upscaler_name: str = "None"
+        self._defer_ratio: float = 1.0
+
+    # ──────────────────────────────────────────────────────────────────────
+    # Метаданные/показ панели
+    # ──────────────────────────────────────────────────────────────────────
+    def title(self):
+        return "Hires. Fix (txt2img + img2img)"
+
+    def show(self, is_img2img):
+        # Делаем доступным и в txt2img, и в img2img
+        return scripts.AlwaysVisible
+
+    # ──────────────────────────────────────────────────────────────────────
+    # UI (Gradio)
+    # ──────────────────────────────────────────────────────────────────────
+    def ui(self, is_img2img):
+        """
+        Порядок возвращаемых компонентов должен соответствовать порядку
+        параметров, которые будет получать setup()/process()/postprocess().
+        """
+        default_upscaler = shared.sd_upscalers[0].name if shared.sd_upscalers else "None"
+        default_ratio = 2.0
+
+        # Удобный дефолт: в img2img работаем "Before", в txt2img — "After"
+        default_after = not is_img2img
+
+        with InputAccordion(value=False, label=self.title()) as enable:
+            with gr.Row():
+                upscaler = gr.Dropdown(
+                    label="Upscaler",
+                    choices=[x.name for x in shared.sd_upscalers] or ["None"],
+                    value=default_upscaler,
+                    scale=3,
+                )
+                ratio = gr.Slider(
+                    minimum=1.0, maximum=8.0, step=0.5,
+                    label="Resize by (scale factor)",
+                    value=default_ratio,
+                    scale=1,
+                )
+
+            with gr.Row():
+                process_after = gr.Checkbox(
+                    label="Process after generation (txt2img-friendly)",
+                    value=default_after,
+                    info="ON: апскейл результата (подходит для txt2img). OFF: апскейл init-изображения (img2img).",
+                )
+                suggest_btn = gr.Button(
+                    value="🔍 Suggest scale from upscaler name",
+                    variant="secondary"
+                )
+
+            info_md = gr.Markdown(value="", visible=True)
+
+            # Авто-под��тановка масштаба из имени апскейлера (2x/4x и т.п.)
+            def _auto_scale_from_name(upscaler_name: str):
+                # Ищем '2x', '4X', 'x3' и т.п.
+                m = re.search(r"(\d)\s*[xX]|[xX]\s*(\d)", (upscaler_name or "").strip())
+                if not m:
+                    return gr.update(), "No explicit scale found in upscaler name."
+                s = int((m.group(1) or m.group(2)))
+                return gr.update(value=float(s)), f"Scale set to **{s}x** from upscaler name."
+
+            upscaler.change(
+                fn=lambda name: _auto_scale_from_name(name)[0],
+                inputs=[upscaler],
+                outputs=[ratio],
+                show_progress="hidden",
+            )
+
+            suggest_btn.click(
+                fn=_auto_scale_from_name,
+                inputs=[upscaler],
+                outputs=[ratio, info_md],
+            )
+
+        # Инфотекст для "Copy params"
+        self.paste_field_names = []
+        self.infotext_fields = [
+            (enable, "Hires. Fix Enabled"),
+            (upscaler, "Hires Upscaler"),
+            (ratio, "Hires Scale"),
+            (process_after, "Hires After Mode"),
+        ]
+        for _, name in self.infotext_fields:
+            self.paste_field_names.append(name)
+
+        # Порядок критичен — именно он попадёт в setup/process/postprocess
+        return [enable, upscaler, ratio, process_after]
+
+    # ──────────────────────────────────────────────────────────────────────
+    # Вспомогательные
+    # ──────────────────────────────────────────────────────────────────────
+    @staticmethod
+    def _get_image_bytes_key(img: Image.Image) -> int:
+        # Стабильный ключ по содержимому изображения
+        try:
+            return hash(img.tobytes())
+        except Exception:
+            return hash(img.convert("RGB").tobytes())
+
+    @staticmethod
+    def _resize_with_upscaler(img: Image.Image, ratio: float, upscaler_name: str) -> Image.Image:
+        # Единая точка апскейла через A1111 helper
+        if ratio <= 1.0 or upscaler_name == "None":
+            return img
+        w, h = img.width, img.height
+        new_w, new_h = int(w * ratio), int(h * ratio)
+        out = images.resize_image(  # resize_mode=0 ("just resize"), но с выбранным апскейлером
+            0, img, new_w, new_h, upscaler_name
+        )
+        return out.convert("RGB")
+
+    # ──────────────────────────────────────────────────────────────────────
+    # setup(): меняем p для img2img (Before) или откладываем на postprocess (After)
+    # ──────────────────────────────────────────────────────────────────────
+    def setup(
+        self,
+        p,
+        enable: bool,
+        upscaler_name: str,
+        ratio: float,
+        process_after: bool,
+        *args,
+        **kwargs,
+    ):
+        # Инфотекст (на стороне p)
+        if getattr(p, "extra_generation_params", None) is not None:
+            p.extra_generation_params.update({
+                "Hires. Fix Enabled": bool(enable),
+                "Hires Upscaler": str(upscaler_name),
+                "Hires Scale": float(ratio),
+                "Hires After Mode": bool(process_after),
+            })
+
+        # Если выключено — ничего не делаем
+        if not enable or upscaler_name == "None":
+            self._defer_after = False
+            return
+
+        # Если есть 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
+            # Ничего не откладываем на postprocess
+            self._defer_after = False
+        else:
+            # txt2img (нет init_images) или пользователь принудил "After"
+            self._defer_after = True
+            self._defer_upscaler_name = upscaler_name
+            self._defer_ratio = float(ratio)
+
+    # ──────────────────────────────────────────────────────────────────────
+    # process(): корректная сигнатура (никаких processed). Нам здесь ничего не нужно.
+    # ──────────────────────────────────────────────────────────────────────
+    def process(
+        self,
+        p,
+        enable: bool,
+        upscaler_name: str,
+        ratio: float,
+        process_after: bool,
+        *args,
+        **kwargs,
+    ):
+        # Ничего не делаем в предгенерационном хуке — вся "Before" логика уже в setup().
+        return
+
+    # ──────────────────────────────────────────────────────────────────────
+    # postprocess(): делаем отложенный апскейл результата (After / txt2img)
+    # ──────────────────────────────────────────────────────────────────────
+    def postprocess(
+        self,
+        p,
+        processed,
+        enable: bool,
+        upscaler_name: str,
+        ratio: float,
+        process_after: bool,
+        *args,
+        **kwargs,
+    ):
+        if not enable:
+            return
+
+        # Выполняем отложенный апскейл результата
+        if self._defer_after:
+            name = self._defer_upscaler_name
+            r = self._defer_ratio
+            new_images = []
+            for img in getattr(processed, "images", []) or []:
+                new_images.append(self._resize_with_upscaler(img, r, name))
+            if new_images:
+                processed.images = new_images
+                if getattr(processed, "extra_generation_params", None) is not None:
+                    processed.extra_generation_params.update({
+                        "Hires Applied After": True
+                    })