fix(api): allow decimal CFG

ssube · Jan 12, 2023 · 2f3b5c0 · 2f3b5c0
1 parent e9553bc
commit 2f3b5c0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/api/onnx_web/serve.py b/api/onnx_web/serve.py
@@ -197,7 +197,7 @@ def pipeline_from_request(pipeline: DiffusionPipeline):
     if negative_prompt == '':
         negative_prompt = None
 
-    cfg = get_and_clamp_int(request.args, 'cfg', default_cfg, config_params.get('cfg').get('max'), 0)
+    cfg = get_and_clamp_float(request.args, 'cfg', default_cfg, config_params.get('cfg').get('max'), 0)
     steps = get_and_clamp_int(request.args, 'steps', default_steps, config_params.get('steps').get('max'))
     height = get_and_clamp_int(
         request.args, 'height', default_height, config_params.get('height').get('max'))