From a3eed7a0958c5029f671caf84d8dbe9b215930ff Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Tue, 31 Jan 2023 21:14:43 -0600
Subject: [PATCH] fix(api): drop unused parameter from inpaint pipeline

---
 api/onnx_web/diffusion/run.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/api/onnx_web/diffusion/run.py b/api/onnx_web/diffusion/run.py
index 848eb5c41..0866f35c5 100644
--- a/api/onnx_web/diffusion/run.py
+++ b/api/onnx_web/diffusion/run.py
@@ -110,7 +110,6 @@ def run_img2img_pipeline(
 
 def run_inpaint_pipeline(
     ctx: ServerContext,
-    stage: StageParams,
     params: ImageParams,
     _size: Size,
     output: str,
@@ -123,6 +122,7 @@ def run_inpaint_pipeline(
     strength: float,
     fill_color: str,
 ) -> None:
+    stage = StageParams()
     image = upscale_outpaint(
         ctx,
         stage,
@@ -142,7 +142,7 @@ def run_inpaint_pipeline(
         logger.info('output image size does not match source, skipping post-blend')
 
     image = run_upscale_correction(
-        ctx, StageParams(), params, image, upscale=upscale)
+        ctx, stage, params, image, upscale=upscale)
 
     dest = base_join(ctx.output_path, output)
     image.save(dest)