diff --git a/api/onnx_web/chain/highres.py b/api/onnx_web/chain/highres.py index 25e541b8..c2166ef3 100644 --- a/api/onnx_web/chain/highres.py +++ b/api/onnx_web/chain/highres.py @@ -43,6 +43,7 @@ def stage_highres( chain.stage( UpscaleSimpleStage(), stage, + method=highres.method, upscale=upscale.with_args(scale=highres.scale, outscale=highres.scale), )