apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): pass outscale factor to CodeFormer (fixes #100)

This commit is contained in:
Sean Sube 2023-02-05 16:22:04 -06:00
parent 85b2895f41
commit ca2f0a6404
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/chain/correct_codeformer.py
View File

@ -15,7 +15,7 @@ device = "cpu"
def correct_codeformer(
job: JobContext,
_server: ServerContext,
_stage: StageParams,
stage: StageParams,
_params: ImageParams,
source: Image.Image,
*,
@ -26,5 +26,5 @@ def correct_codeformer(
# TODO: terrible names, fix
image = source or source_image
pipe = CodeFormer().to(device.torch_device())
pipe = CodeFormer(upscale=stage.outscale).to(device.torch_device())
return pipe(image)