From 3edf5e6c4df91332cd0d4133e5b40bfd41da6763 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sun, 2 Jul 2023 18:57:11 -0500 Subject: [PATCH] fix(api): handle unlimited size stages correctly --- api/onnx_web/chain/base.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/api/onnx_web/chain/base.py b/api/onnx_web/chain/base.py index 23e3e06f5..23976c695 100644 --- a/api/onnx_web/chain/base.py +++ b/api/onnx_web/chain/base.py @@ -155,7 +155,10 @@ def __call__( 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,