apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

pass metadata through tiles

This commit is contained in:
Sean Sube 2024-01-04 09:12:32 -06:00
parent 112d5a8876
commit ce84dfa115
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/onnx_web/chain/blend_img2img.py
View File

@ -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,

2
api/onnx_web/chain/pipeline.py
View File

@ -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,