apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): make sure stage params are declared for all txt2img pipelines

This commit is contained in:
Sean Sube 2023-11-15 23:14:48 -06:00
parent 59685770e4
commit b31227ecb3
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
api/onnx_web/diffusers/run.py
View File

@ -63,8 +63,9 @@ def run_txt2img_pipeline(
# apply upscaling and correction, before highres # apply upscaling and correction, before highres
highres_size = params.unet_tile highres_size = params.unet_tile
stage = StageParams(tile_size=highres_size)
if params.is_panorama(): if params.is_panorama():
stage = StageParams(tile_size=highres_size)
chain.stage( chain.stage(
BlendDenoiseStage(), BlendDenoiseStage(),
stage, stage,