fix strength

ssube · Apr 13, 2023 · eb428e9 · eb428e9
1 parent 8bf8889
commit eb428e9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/api/onnx_web/diffusers/run.py b/api/onnx_web/diffusers/run.py
@@ -240,9 +240,9 @@ def run_img2img_pipeline(
 
     pipe_params = {}
     if params.pipeline == "controlnet":
-        pipe_params["controlnet_conditioning_scale"] = params.strength
+        pipe_params["controlnet_conditioning_scale"] = strength
     elif params.pipeline == "img2img":
-        pipe_params["strength"] = params.strength
+        pipe_params["strength"] = strength
 
     progress = job.get_progress_callback()
     if params.lpw():