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) stage_params.tile_size)
def stage_tile(tile: Image.Image) -> Image.Image: def stage_tile(tile: Image.Image) -> Image.Image:
tile = stage_fn(ctx, stage_params, tile, tile = stage_fn(ctx, stage_params, params, tile,
params, **stage_kwargs) **stage_kwargs)
tile.save(path.join(ctx.output_path, 'last-tile.png')) tile.save(path.join(ctx.output_path, 'last-tile.png'))
return tile return tile
@ -86,10 +86,11 @@ class ChainPipeline:
image, stage_params.tile_size, stage_params.outscale, [stage_tile]) image, stage_params.tile_size, stage_params.outscale, [stage_tile])
else: else:
print('source image within tile size, run stage') print('source image within tile size, run stage')
image = stage_fn(ctx, stage_params, image, image = stage_fn(ctx, stage_params, params, image,
params, **stage_kwargs) **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) print('finished running pipeline, result size: %sx%s' % image.size)
return image 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: def upscale_stage(_ctx: ServerContext, stage: StageParams, params: ImageParams, image: Image.Image) -> Image:
return pipeline( return pipeline(
params.prompt, params.prompt,
image=image, image,
generator=torch.manual_seed(seed), generator=torch.manual_seed(seed),
num_inference_steps=params.steps, num_inference_steps=params.steps,
).images[0] ).images[0]
@ -221,7 +221,7 @@ def run_upscale_correction(
outscale=upscale.outscale) outscale=upscale.outscale)
image = upscale_resrgan(ctx, stage, params, image, upscale=upscale) image = upscale_resrgan(ctx, stage, params, image, upscale=upscale)
elif 'stable-diffusion' in upscale.upscale_model: 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) stage = StageParams(tile_size=mini_tile, outscale=upscale.outscale)
image = upscale_stable_diffusion( image = upscale_stable_diffusion(
ctx, stage, params, image, upscale=upscale) ctx, stage, params, image, upscale=upscale)