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

ssube · Jul 5, 2023 · 3798ea7 · 3798ea7
1 parent fed4315
commit 3798ea7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/api/onnx_web/chain/upscale_stable_diffusion.py b/api/onnx_web/chain/upscale_stable_diffusion.py
@@ -67,6 +67,6 @@ def run(
                 noise_level=upscale.denoise,
                 callback=callback,
             )
-            outputs.extend(result.image)
+            outputs.extend(result.images)
 
         return outputs