1
0
Fork 0

lint(api): split up face correction and upscaling models

This commit is contained in:
Sean Sube 2023-01-16 17:51:32 -06:00
parent decb2813c6
commit dab2cc7752
1 changed files with 4 additions and 1 deletions

View File

@ -98,6 +98,9 @@ mask_filters = {
# TODO: load from model_path
upscale_models = [
'RealESRGAN_x4plus',
]
face_models = [
'GFPGANv1.3',
]
@ -186,7 +189,7 @@ def upscale_from_request() -> UpscaleParams:
scale=scale,
outscale=outscale,
faces=faces,
face_model=upscale_models[1],
face_model=face_models[0],
platform='onnx',
denoise=denoise,
)