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

ssube · Feb 5, 2023 · ca2f0a6 · ca2f0a6
1 parent 85b2895
commit ca2f0a6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/api/onnx_web/chain/correct_codeformer.py b/api/onnx_web/chain/correct_codeformer.py
@@ -15,7 +15,7 @@
 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)