From 6425a91fe3e230ff642b00bd3679b624b76bf9bf Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Tue, 4 Jul 2023 16:24:47 -0500 Subject: [PATCH] fix(api): support multiple outputs from txt2img stage --- api/onnx_web/chain/source_txt2img.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/api/onnx_web/chain/source_txt2img.py b/api/onnx_web/chain/source_txt2img.py index 55c12490..64e75566 100644 --- a/api/onnx_web/chain/source_txt2img.py +++ b/api/onnx_web/chain/source_txt2img.py @@ -88,7 +88,4 @@ class SourceTxt2ImgStage(BaseStage): callback=callback, ) - output = result.images[0] - - logger.info("final output image size: %sx%s", output.width, output.height) - return output + return result.images \ No newline at end of file