From 7c1b85f79357e9f72802a2a3d4f3e64fabc4f462 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Tue, 4 Jul 2023 16:22:19 -0500 Subject: [PATCH] fix(api): define tile size for logging --- api/onnx_web/chain/base.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/api/onnx_web/chain/base.py b/api/onnx_web/chain/base.py index 6ea81567..b1e0f16d 100644 --- a/api/onnx_web/chain/base.py +++ b/api/onnx_web/chain/base.py @@ -140,13 +140,13 @@ class ChainPipeline: ] ) + 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,