fix(api): support multiple outputs from txt2img stage

ssube · Jul 4, 2023 · 6425a91 · 6425a91
1 parent 7c1b85f
commit 6425a91
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/api/onnx_web/chain/source_txt2img.py b/api/onnx_web/chain/source_txt2img.py
@@ -88,7 +88,4 @@ def run(
                 callback=callback,
             )
 
-        output = result.images[0]
-
-        logger.info("final output image size: %sx%s", output.width, output.height)
-        return output
+        return result.images