apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): restore denoise code path for Real ESRGAN v3 (#109)

This commit is contained in:
Sean Sube 2023-02-06 18:11:45 -06:00
parent ad108060ed
commit ae5cf1fd28
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/onnx_web/chain/upscale_resrgan.py
View File

@ -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]