fix(api): separate Real ESRGAN upsampler from GFPGAN stage (#87)

ssube · Feb 12, 2023 · 82487f5 · 82487f5
1 parent ae8aa2e
commit 82487f5
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/api/onnx_web/chain/correct_gfpgan.py b/api/onnx_web/chain/correct_gfpgan.py
@@ -9,7 +9,6 @@
 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,