fix(api): restore denoise code path for Real ESRGAN v3 (#109)
This commit is contained in:
parent
ad108060ed
commit
ae5cf1fd28
|
@ -50,9 +50,9 @@ def load_resrgan(
|
||||||
raise Exception("unknown platform %s" % params.format)
|
raise Exception("unknown platform %s" % params.format)
|
||||||
|
|
||||||
dni_weight = None
|
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(
|
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]
|
model_path = [model_path, wdn_model_path]
|
||||||
dni_weight = [params.denoise, 1 - params.denoise]
|
dni_weight = [params.denoise, 1 - params.denoise]
|
||||||
|
|
Loading…
Reference in New Issue