fix(api): support multiple outputs from txt2img stage

2023-07-04 16:24:47 -05:00 · 2023-07-04 16:24:47 -05:00 · 6425a91fe3
parent 7c1b85f793
commit 6425a91fe3
1 changed files with 1 additions and 4 deletions
--- 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