apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

persist to output dir, do not tile output

This commit is contained in:
Sean Sube 2023-01-28 09:15:02 -06:00
parent 779457b546
commit d89437b11e
2 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -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

4
api/onnx_web/serve.py
View File

@ -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(),
})