From 47bf4c6bb7fb131061a6ca0924a0be17eb6b9bab Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Sat, 14 Jan 2023 23:33:44 -0600
Subject: [PATCH] fix(api): positional args for LPW pipeline

---
 api/onnx_web/serve.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/api/onnx_web/serve.py b/api/onnx_web/serve.py
index 23b3791b..34ea0d09 100644
--- a/api/onnx_web/serve.py
+++ b/api/onnx_web/serve.py
@@ -247,17 +247,17 @@ def run_txt2img_pipeline(model, provider, scheduler, prompt, negative_prompt, cf
     print('saved txt2img output: %s' % (output))
 
 
-def run_img2img_pipeline(model, provider, scheduler, prompt, negative_prompt, cfg, steps, seed, output, strength, input_image):
+def run_img2img_pipeline(model, provider, scheduler, prompt, negative_prompt, cfg, steps, seed, output, strength, source_image):
     pipe = load_pipeline(OnnxStableDiffusionImg2ImgPipeline,
                          model, provider, scheduler)
 
     rng = np.random.RandomState(seed)
 
     image = pipe.img2img(
+        source_image,
         prompt,
         generator=rng,
         guidance_scale=cfg,
-        image=input_image,
         negative_prompt=negative_prompt,
         num_inference_steps=steps,
         strength=strength,
@@ -280,13 +280,13 @@ def run_inpaint_pipeline(model, provider, scheduler, prompt, negative_prompt, cf
             source_image, mask_image, (left, right, top, bottom))
 
     image = pipe.inpaint(
+        source_image,
+        mask_image,
         prompt,
         generator=rng,
         guidance_scale=cfg,
         height=height,
-        image=source_image,
         latents=latents,
-        mask_image=mask_image,
         negative_prompt=negative_prompt,
         num_inference_steps=steps,
         width=width,