fix(api): remove unintentional tuple in img2img pipe

ssube · Jul 5, 2023 · fed4315 · fed4315
1 parent c0dd4e9
commit fed4315
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/api/onnx_web/diffusers/run.py b/api/onnx_web/diffusers/run.py
@@ -171,7 +171,7 @@ def run_img2img_pipeline(
 
     # run and append the filtered source
     progress = job.get_progress_callback()
-    images = chain(job, server, params, [source], callback=progress),
+    images = chain(job, server, params, [source], callback=progress)
 
     if source_filter is not None and source_filter != "none":
         images.append(source)

diff --git a/api/onnx_web/diffusers/utils.py b/api/onnx_web/diffusers/utils.py
@@ -312,13 +312,17 @@ def parse_prompt(
     params: ImageParams,
     use_input: bool = False,
 ) -> Tuple[List[Tuple[str, str]], List[Tuple[str, float]], List[Tuple[str, float]]]:
-    prompt, loras = get_loras_from_prompt(params.input_prompt if use_input else params.prompt)
+    prompt, loras = get_loras_from_prompt(
+        params.input_prompt if use_input else params.prompt
+    )
     prompt, inversions = get_inversions_from_prompt(prompt)
     params.prompt = prompt
 
     neg_prompt = None
     if params.input_negative_prompt is not None:
-        neg_prompt, neg_loras = get_loras_from_prompt(params.input_negative_prompt if use_input else params.negative_prompt)
+        neg_prompt, neg_loras = get_loras_from_prompt(
+            params.input_negative_prompt if use_input else params.negative_prompt
+        )
         neg_prompt, neg_inversions = get_inversions_from_prompt(neg_prompt)
         params.negative_prompt = neg_prompt