fix(api): collect final images from SD upscale stage

2023-07-05 08:22:29 -05:00 · 2023-07-05 08:22:29 -05:00 · 3798ea77e7
parent fed4315095
commit 3798ea77e7
1 changed files with 1 additions and 1 deletions
--- a/api/onnx_web/chain/upscale_stable_diffusion.py
+++ b/api/onnx_web/chain/upscale_stable_diffusion.py
@ -67,6 +67,6 @@ class UpscaleStableDiffusionStage(BaseStage):
                noise_level=upscale.denoise,
                callback=callback,
            )
-            outputs.extend(result.image)
+            outputs.extend(result.images)

        return outputs