fix(api): define tile size for logging

ssube · Jul 4, 2023 · 7c1b85f · 7c1b85f
1 parent 99a073a
commit 7c1b85f
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/api/onnx_web/chain/base.py b/api/onnx_web/chain/base.py
@@ -140,13 +140,13 @@ def __call__(
                 ]
             )
 
+            tile = stage_params.tile_size
+            if stage_pipe.max_tile > 0:
+                tile = min(stage_pipe.max_tile, stage_params.tile_size)
+
             if must_tile:
                 stage_outputs = []
                 for source in stage_sources:
-                    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,