From 66e938f9ab247d5d9ca21021f9004d7be0346d26 Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Fri, 31 Mar 2023 23:32:05 -0500
Subject: [PATCH] split steps before and after highres

---
 api/onnx_web/diffusers/run.py | 19 ++++++++++---------
 gui/Makefile                  |  1 +
 2 files changed, 11 insertions(+), 9 deletions(-)

diff --git a/api/onnx_web/diffusers/run.py b/api/onnx_web/diffusers/run.py
index cee20be6..a5e04e96 100644
--- a/api/onnx_web/diffusers/run.py
+++ b/api/onnx_web/diffusers/run.py
@@ -30,6 +30,11 @@ def run_txt2img_pipeline(
     outputs: List[str],
     upscale: UpscaleParams,
 ) -> None:
+    # TODO: add to params
+    highres_scale = 4
+    highres_steps = 15
+    highres_strength = 0.5
+
     latents = get_latents_from_seed(params.seed, size, batch=params.batch)
 
     (prompt, loras) = get_loras_from_prompt(params.prompt)
@@ -60,7 +65,7 @@ def run_txt2img_pipeline(
             latents=latents,
             negative_prompt=params.negative_prompt,
             num_images_per_prompt=params.batch,
-            num_inference_steps=params.steps,
+            num_inference_steps=highres_steps,
             eta=params.eta,
             callback=progress,
         )
@@ -75,17 +80,13 @@ def run_txt2img_pipeline(
             latents=latents,
             negative_prompt=params.negative_prompt,
             num_images_per_prompt=params.batch,
-            num_inference_steps=params.steps,
+            num_inference_steps=highres_steps,
             eta=params.eta,
             callback=progress,
         )
 
     for image, output in zip(result.images, outputs):
-        highres_scale = 4
-        highres_strength = 0.5
-
-        if params.highres > 1:
-
+        if highres_scale > 1:
             def highres(tile: Image.Image, dims):
                 highpipe = load_pipeline(
                     server,
@@ -108,7 +109,7 @@ def run_txt2img_pipeline(
                         guidance_scale=params.cfg,
                         negative_prompt=params.negative_prompt,
                         num_images_per_prompt=1,
-                        num_inference_steps=params.steps,
+                        num_inference_steps=params.steps - highres_steps,
                         strength=highres_strength,
                         eta=params.eta,
                         callback=progress,
@@ -123,7 +124,7 @@ def run_txt2img_pipeline(
                         guidance_scale=params.cfg,
                         negative_prompt=params.negative_prompt,
                         num_images_per_prompt=1,
-                        num_inference_steps=params.steps,
+                        num_inference_steps=params.steps - highres_steps,
                         strength=highres_strength,
                         eta=params.eta,
                         callback=progress,
diff --git a/gui/Makefile b/gui/Makefile
index 676cabd5..8f86f0b0 100644
--- a/gui/Makefile
+++ b/gui/Makefile
@@ -35,6 +35,7 @@ bundle: build
 COVER_OPTS := --all \
 		--exclude ".eslintrc.js" \
 		--exclude "docs/**" \
+    --exclude "out/bundle/**" \
 		--exclude "out/coverage/**" \
 		--exclude "vendor/**" \
 		--reporter=text-summary \