apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): remove unintentional tuple in img2img pipe

This commit is contained in:
Sean Sube 2023-07-05 06:50:33 -05:00
parent c0dd4e96dc
commit fed4315095
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/onnx_web/diffusers/run.py
View File

@ -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)

8
api/onnx_web/diffusers/utils.py
View File

@ -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