From d89437b11eef123decb993206871be6d0724e32a Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Sat, 28 Jan 2023 09:15:02 -0600
Subject: [PATCH] persist to output dir, do not tile output

---
 api/onnx_web/chain/persist_disk.py | 6 ++++--
 api/onnx_web/serve.py              | 4 ++--
 2 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/api/onnx_web/chain/persist_disk.py b/api/onnx_web/chain/persist_disk.py
index c10bf845..cbfd1e79 100644
--- a/api/onnx_web/chain/persist_disk.py
+++ b/api/onnx_web/chain/persist_disk.py
@@ -6,6 +6,7 @@ from ..params import (
     StageParams,
 )
 from ..utils import (
+    base_join,
     ServerContext,
 )
 
@@ -18,6 +19,7 @@ def persist_disk(
     *,
     output: str,
 ) -> Image.Image:
-    source_image.save(output)
-    print('saved image to %s' % (output,))
+    dest = base_join(ctx.output_path, output)
+    source_image.save(dest)
+    print('saved image to %s' % (dest,))
     return source_image
diff --git a/api/onnx_web/serve.py b/api/onnx_web/serve.py
index e5397bd8..2cc4dfee 100644
--- a/api/onnx_web/serve.py
+++ b/api/onnx_web/serve.py
@@ -552,7 +552,7 @@ def chain():
         (upscale_outpaint, StageParams(outscale=4), {
             'expand': Border(256, 256, 256, 256),
         }),
-        (persist_disk, StageParams(), {
+        (persist_disk, StageParams(tile_size=8192), {
             'output': output,
         })
     ])
@@ -563,7 +563,7 @@ def chain():
     return jsonify({
         'output': output,
         'params': params.tojson(),
-        'size': upscale.resize(size).tojson(),
+        'size': size.tojson(),
     })