fix parameter order for chain pipeline
This commit is contained in:
parent
bfd8e0a739
commit
9b37c174d1
|
@ -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
|
||||
|
|
|
@ -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)
|
||||
|
|
Loading…
Reference in New Issue