apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): remove inpaint post-multiply (#150)

This commit is contained in:
Sean Sube 2023-02-15 17:45:25 -06:00
parent df5fb93bc0
commit b5eeb44c62
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 4 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
api/logging.yaml
View File

@ -5,14 +5,14 @@ formatters:
handlers: handlers:
console: console:
class: logging.StreamHandler class: logging.StreamHandler
level: DEBUG level: INFO
formatter: simple formatter: simple
stream: ext://sys.stdout stream: ext://sys.stdout
loggers: loggers:
'': '':
level: DEBUG level: INFO
handlers: [console] handlers: [console]
propagate: True propagate: True
root: root:
level: DEBUG level: INFO
handlers: [console] handlers: [console]

7
api/onnx_web/diffusion/run.py
View File

@ -177,6 +177,7 @@ def run_inpaint_pipeline(
# calling the upscale_outpaint stage directly needs accumulating progress # calling the upscale_outpaint stage directly needs accumulating progress
progress = ChainProgress.from_progress(progress) progress = ChainProgress.from_progress(progress)
logger.debug("applying mask filter and generating noise source")
image = upscale_outpaint( image = upscale_outpaint(
job, job,
server, server,
@ -190,12 +191,6 @@ def run_inpaint_pipeline(
noise_source=noise_source, noise_source=noise_source,
callback=progress, callback=progress,
) )
logger.info("applying mask filter and generating noise source")
if image.size == source_image.size:
image = ImageChops.blend(source_image, image, strength)
else:
logger.info("output image size does not match source, skipping post-blend")
image = run_upscale_correction( image = run_upscale_correction(
job, server, stage, params, image, upscale=upscale, callback=progress job, server, stage, params, image, upscale=upscale, callback=progress