apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix parameter order for chain pipeline

This commit is contained in:
Sean Sube 2023-01-27 17:26:48 -06:00
parent bfd8e0a739
commit 9b37c174d1
2 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
api/onnx_web/chain.py
View File

@ -77,8 +77,8 @@ class ChainPipeline:
stage_params.tile_size)
def stage_tile(tile: Image.Image) -> Image.Image:
tile = stage_fn(ctx, stage_params, tile,
params, **stage_kwargs)
tile = stage_fn(ctx, stage_params, params, tile,
**stage_kwargs)
tile.save(path.join(ctx.output_path, 'last-tile.png'))
return tile
@ -86,10 +86,11 @@ class ChainPipeline:
image, stage_params.tile_size, stage_params.outscale, [stage_tile])
else:
print('source image within tile size, run stage')
image = stage_fn(ctx, stage_params, image,
params, **stage_kwargs)
image = stage_fn(ctx, stage_params, params, image,
**stage_kwargs)
print('finished running pipeline stage, result size: %sx%s' % image.size)
print('finished running pipeline stage, result size: %sx%s' %
image.size)
print('finished running pipeline, result size: %sx%s' % image.size)
return image

4
api/onnx_web/upscale.py
View File

@ -194,7 +194,7 @@ def upscale_stable_diffusion(
def upscale_stage(_ctx: ServerContext, stage: StageParams, params: ImageParams, image: Image.Image) -> Image:
return pipeline(
params.prompt,
image=image,
image,
generator=torch.manual_seed(seed),
num_inference_steps=params.steps,
).images[0]
@ -221,7 +221,7 @@ def run_upscale_correction(
outscale=upscale.outscale)
image = upscale_resrgan(ctx, stage, params, image, upscale=upscale)
elif 'stable-diffusion' in upscale.upscale_model:
mini_tile = max(128, stage.tile_size)
mini_tile = min(128, stage.tile_size)
stage = StageParams(tile_size=mini_tile, outscale=upscale.outscale)
image = upscale_stable_diffusion(
ctx, stage, params, image, upscale=upscale)