From d5e55889399f0a641c7c32ac953bf2b801dedb0e Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Mon, 30 Jan 2023 20:54:39 -0600 Subject: [PATCH] fix(api): supply missing parameter to GFPGAN --- api/onnx_web/chain/correct_gfpgan.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/api/onnx_web/chain/correct_gfpgan.py b/api/onnx_web/chain/correct_gfpgan.py index b24970cb..07bc318e 100644 --- a/api/onnx_web/chain/correct_gfpgan.py +++ b/api/onnx_web/chain/correct_gfpgan.py @@ -24,7 +24,7 @@ last_pipeline_instance = None last_pipeline_params = None -def load_gfpgan(ctx: ServerContext, upscale: UpscaleParams): +def load_gfpgan(ctx: ServerContext, upscale: UpscaleParams, upsampler: Optional[RealESRGANer] = None): global last_pipeline_instance global last_pipeline_params @@ -67,7 +67,7 @@ def correct_gfpgan( return image logger.info('correcting faces with GFPGAN model: %s', upscale.correction_model) - gfpgan = load_gfpgan(ctx, upscale) + gfpgan = load_gfpgan(ctx, upscale, upsampler=upsampler) _, _, output = gfpgan.enhance( image, has_aligned=False, only_center_face=False, paste_back=True, weight=upscale.face_strength)