1
0
Fork 0

only add denoise stage after panorama pipeline

This commit is contained in:
Sean Sube 2023-11-12 21:19:01 -06:00
parent 35171e6f12
commit 21d1240e28
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 9 additions and 9 deletions

View File

@ -62,17 +62,17 @@ def run_txt2img_pipeline(
)
# apply upscaling and correction, before highres
if params.is_panorama() and server.panorama_tiles:
highres_size = tile_size * highres.scale
else:
highres_size = params.unet_tile
if params.is_panorama():
stage = StageParams(tile_size=highres_size)
chain.stage(
BlendDenoiseStage(),
stage,
)
if server.panorama_tiles:
highres_size = tile_size * highres.scale
first_upscale, after_upscale = split_upscale(upscale)
if first_upscale:
stage_upscale_correction(