apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

save mask and noise for debugging

This commit is contained in:
Sean Sube 2023-01-15 10:09:57 -06:00
parent 6c07d124e0
commit 4af1b8e1aa
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
api/onnx_web/serve.py
View File

@ -304,13 +304,18 @@ def run_inpaint_pipeline(
if left > 0 or right > 0 or top > 0 or bottom > 0:
print('expanding image for outpainting')
source_image, mask_image, _full_noise, _full_dims = expand_image(
source_image, mask_image, noise_image, _full_dims = expand_image(
source_image,
mask_image,
(left, right, top, bottom),
noise_source=noise_source,
blend_op=blend_op)
if environ.get('DEBUG') is not None:
source_image.save('./last-source.png')
mask_image.save('./last-mask.png')
noise_image.save('./last-noise.png')
image = pipe(
prompt,
generator=rng,