feat(api): apply overlap to highres upscaling stage

ssube · Jul 3, 2023 · ed953f5 · ed953f5
1 parent 47b1094
commit ed953f5
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/api/onnx_web/chain/highres.py b/api/onnx_web/chain/highres.py
@@ -42,13 +42,15 @@ def stage_highres(
                     outscale=highres.scale,
                 ),
                 chain=chain,
+                overlap=params.overlap,
             )
         else:
             logger.debug("using simple upscaling for highres")
             chain.stage(
                 UpscaleSimpleStage(),
                 stage,
                 method=highres.method,
+                overlap=params.overlap,
                 upscale=upscale.with_args(scale=highres.scale, outscale=highres.scale),
             )