save mask and noise for debugging
This commit is contained in:
parent
6c07d124e0
commit
4af1b8e1aa
|
@ -304,13 +304,18 @@ def run_inpaint_pipeline(
|
||||||
|
|
||||||
if left > 0 or right > 0 or top > 0 or bottom > 0:
|
if left > 0 or right > 0 or top > 0 or bottom > 0:
|
||||||
print('expanding image for outpainting')
|
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,
|
source_image,
|
||||||
mask_image,
|
mask_image,
|
||||||
(left, right, top, bottom),
|
(left, right, top, bottom),
|
||||||
noise_source=noise_source,
|
noise_source=noise_source,
|
||||||
blend_op=blend_op)
|
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(
|
image = pipe(
|
||||||
prompt,
|
prompt,
|
||||||
generator=rng,
|
generator=rng,
|
||||||
|
|
Loading…
Reference in New Issue