From fed43150959c2c23c7e91abeb135c7e3e13d5776 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Wed, 5 Jul 2023 06:50:33 -0500 Subject: [PATCH] fix(api): remove unintentional tuple in img2img pipe --- api/onnx_web/diffusers/run.py | 2 +- api/onnx_web/diffusers/utils.py | 8 ++++++-- 2 files changed, 7 insertions(+), 3 deletions(-) diff --git a/api/onnx_web/diffusers/run.py b/api/onnx_web/diffusers/run.py index c72bc01c..d6a7db3e 100644 --- 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 index 04a0bf69..fc45fdff 100644 --- 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