apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): keep strength as a float

This commit is contained in:
Sean Sube 2023-01-07 15:19:24 -06:00
parent 98a8db16a3
commit 926f77b3b0
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/onnx_web/serve.py
View File

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