From 82487f5771895f581cad5b5144ee84a0db274175 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sun, 12 Feb 2023 13:50:28 -0600 Subject: [PATCH] fix(api): separate Real ESRGAN upsampler from GFPGAN stage (#87) --- api/onnx_web/chain/correct_gfpgan.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/api/onnx_web/chain/correct_gfpgan.py b/api/onnx_web/chain/correct_gfpgan.py index 2f19c98b..a02d87ae 100644 --- a/api/onnx_web/chain/correct_gfpgan.py +++ b/api/onnx_web/chain/correct_gfpgan.py @@ -9,7 +9,6 @@ from PIL import Image from ..device_pool import JobContext from ..params import DeviceParams, ImageParams, StageParams, UpscaleParams from ..utils import ServerContext, run_gc -from .upscale_resrgan import load_resrgan logger = getLogger(__name__) @@ -35,12 +34,10 @@ def load_gfpgan( logger.debug("loading GFPGAN model from %s", face_path) - upsampler = load_resrgan(server, upscale, device, tile=stage.tile_size) - # TODO: find a way to pass the ONNX model to underlying architectures gfpgan = GFPGANer( arch="clean", - bg_upsampler=upsampler, + bg_upsampler=None, channel_multiplier=2, model_path=face_path, upscale=upscale.face_outscale,