apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

Merge pull request #411 from HoopyFreud/main 

Fixed inpaint issue with no tile_mask
This commit is contained in:
Sean Sube 2023-11-12 16:40:26 -06:00 committed by GitHub
parent 3ffbc00390 63dfec6770
commit bdda39f924
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -160,11 +160,11 @@ class ChainPipeline:
if stage_pipe.max_tile > 0: if stage_pipe.max_tile > 0:
tile = min(stage_pipe.max_tile, stage_params.tile_size) tile = min(stage_pipe.max_tile, stage_params.tile_size)
if must_tile: if stage_sources or must_tile:
stage_outputs = [] stage_outputs = []
for source in stage_sources: for source in stage_sources:
logger.info( logger.info(
"image larger than tile size of %s, tiling stage", "image contains sources or is larger than tile size of %s, tiling stage",
tile, tile,
) )
@ -232,7 +232,7 @@ class ChainPipeline:
stage_sources = stage_outputs stage_sources = stage_outputs
else: else:
logger.debug("image within tile size of %s, running stage", tile) logger.debug("image does not contain sources and is within tile size of %s, running stage", tile)
for i in range(worker.retries): for i in range(worker.retries):
try: try:
stage_outputs = stage_pipe.run( stage_outputs = stage_pipe.run(