fix(api): drop unused parameter from inpaint pipeline

ssube · Feb 1, 2023 · a3eed7a · a3eed7a
1 parent 6a0f881
commit a3eed7a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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)