fix(api): resize blend mask to match source

2023-12-20 23:41:54 -06:00 · 2023-12-20 23:41:54 -06:00 · 2c2eda8c3c
parent 1fbee0ae52
commit 2c2eda8c3c
1 changed files with 6 additions and 2 deletions
--- a/api/onnx_web/diffusers/run.py
+++ b/api/onnx_web/diffusers/run.py
@ -527,11 +527,15 @@ def run_blend_pipeline(
    chain = ChainPipeline()
    tile_size = get_base_tile(params, size)

+    # resize mask to match source size
+    stage_source = sources[1]
+    stage_mask = mask.resize(stage_source.size, Image.Resampling.BILINEAR)
+
    chain.stage(
        BlendMaskStage(),
        StageParams(tile_size=tile_size),
-        stage_source=sources[1],
-        stage_mask=mask,
+        stage_source=stage_source,
+        stage_mask=stage_mask,
    )

    # apply upscaling and correction