fix(api): handle panoramic outpainting output

ssube · Jul 5, 2023 · c0dd4e9 · c0dd4e9
1 parent 0b5f554
commit c0dd4e9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/api/onnx_web/chain/upscale_outpaint.py b/api/onnx_web/chain/upscale_outpaint.py
@@ -138,7 +138,7 @@ def outpaint(tile_source: Image.Image, dims: Tuple[int, int, int]):
 
             if params.pipeline == "panorama":
                 logger.debug("outpainting with one shot panorama, no tiling")
-                return outpaint(source, (0, 0, max(source.width, source.height)))
+                output = outpaint(source, (0, 0, max(source.width, source.height)))
             if overlap == 0:
                 logger.debug("outpainting with 0 margin, using grid tiling")
                 output = process_tile_grid(source, SizeChart.auto, 1, [outpaint])