From ae5cf1fd28d05587e3d942d045a41241e2af3a18 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Mon, 6 Feb 2023 18:11:45 -0600 Subject: [PATCH] fix(api): restore denoise code path for Real ESRGAN v3 (#109) --- api/onnx_web/chain/upscale_resrgan.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/api/onnx_web/chain/upscale_resrgan.py b/api/onnx_web/chain/upscale_resrgan.py index e22522be..25835ef7 100644 --- a/api/onnx_web/chain/upscale_resrgan.py +++ b/api/onnx_web/chain/upscale_resrgan.py @@ -50,9 +50,9 @@ def load_resrgan( raise Exception("unknown platform %s" % params.format) dni_weight = None - if params.upscale_model == "realesr-general-x4v3" and params.denoise != 1: + if params.upscale_model == "real-esrgan-x4-v3" and params.denoise != 1: wdn_model_path = model_path.replace( - "realesr-general-x4v3", "realesr-general-wdn-x4v3" + "real-esrgan-x4-v3", "real-esrgan-x4-v3-wdn" ) model_path = [model_path, wdn_model_path] dni_weight = [params.denoise, 1 - params.denoise]