apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): pass upscale params to correction stages

This commit is contained in:
Sean Sube 2023-06-30 22:33:14 -05:00
parent 94688ab3c7
commit 5556478669
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/onnx_web/diffusers/upscale.py
View File

@ -98,9 +98,9 @@ def stage_upscale_correction(
tile_size=stage.tile_size, outscale=upscale.face_outscale
)
if "codeformer" in upscale.correction_model:
correct_stage = (correct_codeformer, face_params, None)
correct_stage = (correct_codeformer, face_params, upscale_opts)
elif "gfpgan" in upscale.correction_model:
correct_stage = (correct_gfpgan, face_params, None)
correct_stage = (correct_gfpgan, face_params, upscale_opts)
else:
logger.warn("unknown correction model: %s", upscale.correction_model)