apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix controlnet strength param

This commit is contained in:
Sean Sube 2023-04-12 23:30:59 -05:00
parent c91bdb7679
commit 8bf888959f
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -237,6 +237,13 @@ def run_img2img_pipeline(
inversions=inversions,
loras=loras,
)
pipe_params = {}
if params.pipeline == "controlnet":
pipe_params["controlnet_conditioning_scale"] = params.strength
elif params.pipeline == "img2img":
pipe_params["strength"] = params.strength
progress = job.get_progress_callback()
if params.lpw():
logger.debug("using LPW pipeline for img2img")
@ -249,9 +256,9 @@ def run_img2img_pipeline(
negative_prompt=params.negative_prompt,
num_images_per_prompt=params.batch,
num_inference_steps=params.steps,
strength=strength,
eta=params.eta,
callback=progress,
**pipe_params,
)
else:
rng = np.random.RandomState(params.seed)
@ -263,9 +270,9 @@ def run_img2img_pipeline(
negative_prompt=params.negative_prompt,
num_images_per_prompt=params.batch,
num_inference_steps=params.steps,
strength=strength,
eta=params.eta,
callback=progress,
**pipe_params,
)
for image, output in zip(result.images, outputs):