apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

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

This commit is contained in:
Sean Sube 2023-09-24 10:11:52 -05:00
parent 539140909b
commit 5ffe8b26fa
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/onnx_web/chain/blend_grid.py
View File

@ -3,7 +3,7 @@ from typing import List, Optional
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 @@ logger = getLogger(__name__)
class BlendGridStage(BaseStage):
max_tile = SizeChart.max
def run(
self,
_worker: WorkerContext,

4
api/onnx_web/chain/persist_disk.py
View File

@ -4,7 +4,7 @@ from typing import List, Optional
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 @@ logger = getLogger(__name__)
class PersistDiskStage(BaseStage):
max_tile = SizeChart.max
def run(
self,
_worker: WorkerContext,