fix(api): resize blend mask to match source

ssube · Dec 21, 2023 · 2c2eda8 · 2c2eda8
1 parent 1fbee0a
commit 2c2eda8
Showing 1 changed file with 6 additions and 2 deletions.
diff --git 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