fix(api): increase max tile size for output and grid stages to max

ssube · Sep 24, 2023 · 5ffe8b2 · 5ffe8b2
1 parent 5391409
commit 5ffe8b2
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/api/onnx_web/chain/blend_grid.py b/api/onnx_web/chain/blend_grid.py
@@ -3,7 +3,7 @@
 
 from PIL import Image
 
-from ..params import ImageParams, StageParams
+from ..params import ImageParams, SizeChart, StageParams
 from ..server import ServerContext
 from ..worker import ProgressCallback, WorkerContext
 from .stage import BaseStage
@@ -12,6 +12,8 @@
 
 
 class BlendGridStage(BaseStage):
+    max_tile = SizeChart.max
+
     def run(
         self,
         _worker: WorkerContext,

diff --git a/api/onnx_web/chain/persist_disk.py b/api/onnx_web/chain/persist_disk.py
@@ -4,7 +4,7 @@
 from PIL import Image
 
 from ..output import save_image
-from ..params import ImageParams, Size, StageParams
+from ..params import ImageParams, Size, SizeChart, StageParams
 from ..server import ServerContext
 from ..worker import WorkerContext
 from .stage import BaseStage
@@ -13,6 +13,8 @@
 
 
 class PersistDiskStage(BaseStage):
+    max_tile = SizeChart.max
+
     def run(
         self,
         _worker: WorkerContext,