diff --git a/src/corridorkey_mlx/engine.py b/src/corridorkey_mlx/engine.py index 5170bd6..2c5b9a9 100644 --- a/src/corridorkey_mlx/engine.py +++ b/src/corridorkey_mlx/engine.py @@ -100,10 +100,14 @@ class CorridorKeyMLXEngine: self._overlap = overlap if self._tiled: - # Tiled: model runs at tile_size, input stays full-res + # Tiled: model runs at tile_size, input stays full-res. + # Tiles are fixed-shape, so fused compilation applies to them the + # same as full-frame — honor the caller's ``compile`` flag rather + # than forcing it off (measured ~1.3-1.45x faster tiled inference + # on M-series Ultras at identical output). self._img_size = self._tile_size self._model: GreenFormer = load_model( - checkpoint, img_size=self._tile_size, compile=False, slim=True + checkpoint, img_size=self._tile_size, compile=compile, slim=True ) logger.info( "Tiled inference: tile_size=%d, overlap=%d", self._tile_size, self._overlap