From 926f77b3b0758a16196392a505d5ca695e54422b Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sat, 7 Jan 2023 15:19:24 -0600 Subject: [PATCH] fix(api): keep strength as a float --- api/onnx_web/serve.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/api/onnx_web/serve.py b/api/onnx_web/serve.py index 5ca0a4a3..ae1cd8ca 100644 --- a/api/onnx_web/serve.py +++ b/api/onnx_web/serve.py @@ -222,7 +222,9 @@ def img2img(): input_image = Image.open(BytesIO(input_file.read())).convert('RGB') input_image.thumbnail((default_width, default_height)) - strength = get_and_clamp(request.args, 'strength', 1.0, 1.0, 0.0) + strength = float(request.args.get('strength', 0.8)) + print('strength: %s' % (strength)) + (model, provider, scheduler, prompt, cfg, steps, height, width, seed, pipe) = pipeline_from_request(OnnxStableDiffusionImg2ImgPipeline)