apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): handle unlimited size stages correctly

This commit is contained in:
Sean Sube 2023-07-02 18:57:11 -05:00
parent d8ec93a619
commit 3edf5e6c4d
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
api/onnx_web/chain/base.py
View File

@ -155,7 +155,10 @@ class ChainPipeline:
size=kwargs.get("size", None),
source=image,
):
tile = min(stage_pipe.max_tile, stage_params.tile_size)
tile = stage_params.tile_size
if stage_pipe.max_tile > 0:
tile = min(stage_pipe.max_tile, stage_params.tile_size)
logger.info(
"image larger than tile size of %s, tiling stage",
tile,