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)
|
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
|
||||||
|
|
|
@ -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)
|
||||||
|
|
Loading…
Reference in New Issue