fix(api): correct order for img2img params

ssube · Feb 6, 2023 · 6fcfe4f · 6fcfe4f
1 parent 35445ff
commit 6fcfe4f
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 5 deletions.
diff --git a/api/onnx_web/diffusion/load.py b/api/onnx_web/diffusion/load.py
@@ -1,5 +1,5 @@
 from logging import getLogger
-from typing import Any, Tuple
+from typing import Any, Optional, Tuple
 
 import numpy as np
 from diffusers import DiffusionPipeline
@@ -9,9 +9,15 @@
 
 logger = getLogger(__name__)
 
-last_pipeline_instance = None
-last_pipeline_options = (None, None, None)
-last_pipeline_scheduler = None
+last_pipeline_instance: Any = None
+last_pipeline_options: Tuple[
+    Optional[DiffusionPipeline],
+    Optional[str],
+    Optional[str],
+    Optional[str],
+    Optional[bool],
+] = (None, None, None, None, None)
+last_pipeline_scheduler: Any = None
 
 latent_channels = 4
 latent_factor = 8

diff --git a/api/onnx_web/diffusion/run.py b/api/onnx_web/diffusion/run.py
@@ -111,8 +111,8 @@ def run_img2img_pipeline(
     else:
         rng = np.random.RandomState(params.seed)
         result = pipe(
-            source_image,
             params.prompt,
+            source_image,
             generator=rng,
             guidance_scale=params.cfg,
             negative_prompt=params.negative_prompt,