lint(api): split up face correction and upscaling models
This commit is contained in:
parent
decb2813c6
commit
dab2cc7752
|
@ -98,6 +98,9 @@ mask_filters = {
|
||||||
# TODO: load from model_path
|
# TODO: load from model_path
|
||||||
upscale_models = [
|
upscale_models = [
|
||||||
'RealESRGAN_x4plus',
|
'RealESRGAN_x4plus',
|
||||||
|
]
|
||||||
|
|
||||||
|
face_models = [
|
||||||
'GFPGANv1.3',
|
'GFPGANv1.3',
|
||||||
]
|
]
|
||||||
|
|
||||||
|
@ -186,7 +189,7 @@ def upscale_from_request() -> UpscaleParams:
|
||||||
scale=scale,
|
scale=scale,
|
||||||
outscale=outscale,
|
outscale=outscale,
|
||||||
faces=faces,
|
faces=faces,
|
||||||
face_model=upscale_models[1],
|
face_model=face_models[0],
|
||||||
platform='onnx',
|
platform='onnx',
|
||||||
denoise=denoise,
|
denoise=denoise,
|
||||||
)
|
)
|
||||||
|
|
Loading…
Reference in New Issue