fix(api): handle unlimited size stages correctly

ssube · Jul 2, 2023 · 3edf5e6 · 3edf5e6
1 parent d8ec93a
commit 3edf5e6
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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,