diff --git a/api/onnx_web/chain/blend_img2img.py b/api/onnx_web/chain/blend_img2img.py index 4528946f..2c838c73 100644 --- a/api/onnx_web/chain/blend_img2img.py +++ b/api/onnx_web/chain/blend_img2img.py @@ -102,7 +102,7 @@ class BlendImg2ImgStage(BaseStage): outputs.extend(result.images) - return StageResult(images=outputs) + return StageResult(images=outputs, metadata=sources.metadata) def steps( self, diff --git a/api/onnx_web/chain/pipeline.py b/api/onnx_web/chain/pipeline.py index 8d6c6816..6c9451c1 100644 --- a/api/onnx_web/chain/pipeline.py +++ b/api/onnx_web/chain/pipeline.py @@ -188,7 +188,7 @@ class ChainPipeline: server, stage_params, per_stage_params, - StageResult(images=source_tile), + StageResult(images=source_tile, metadata=stage_sources.metadata), tile_mask=tile_mask, callback=callback, dims=dims,