From ce84dfa1150c7f67b6a19e7bfd560bd0f98a65bd Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Thu, 4 Jan 2024 09:12:32 -0600 Subject: [PATCH] pass metadata through tiles --- api/onnx_web/chain/blend_img2img.py | 2 +- api/onnx_web/chain/pipeline.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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,