add size
This commit is contained in:
parent
b0df6247b2
commit
0047e1a12c
|
@ -4,7 +4,7 @@ from typing import List, Optional
|
|||
from PIL import Image
|
||||
|
||||
from ..output import save_image
|
||||
from ..params import ImageParams, StageParams
|
||||
from ..params import ImageParams, Size, StageParams
|
||||
from ..server import ServerContext
|
||||
from ..worker import WorkerContext
|
||||
from .stage import BaseStage
|
||||
|
@ -22,11 +22,12 @@ class PersistDiskStage(BaseStage):
|
|||
sources: List[Image.Image],
|
||||
*,
|
||||
output: List[str],
|
||||
size: Optional[Size] = None,
|
||||
stage_source: Optional[Image.Image] = None,
|
||||
**kwargs,
|
||||
) -> List[Image.Image]:
|
||||
for source, name in zip(sources, output):
|
||||
dest = save_image(server, name, source, params=params)
|
||||
dest = save_image(server, name, source, params=params, size=size)
|
||||
logger.info("saved image to %s", dest)
|
||||
|
||||
return sources
|
||||
|
|
Loading…
Reference in New Issue