From ca2f0a6404c86e0162671fcf31817803485664ea Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Sun, 5 Feb 2023 16:22:04 -0600
Subject: [PATCH] fix(api): pass outscale factor to CodeFormer (fixes #100)

---
 api/onnx_web/chain/correct_codeformer.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/api/onnx_web/chain/correct_codeformer.py b/api/onnx_web/chain/correct_codeformer.py
index 9d80b83a..60bea425 100644
--- a/api/onnx_web/chain/correct_codeformer.py
+++ b/api/onnx_web/chain/correct_codeformer.py
@@ -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)