This commit is contained in:
Harisreedhar 2024-10-29 21:47:25 +05:30 committed by henryruhs
parent f75410e1e1
commit 5c9d893dab
3 changed files with 12 additions and 12 deletions

View File

@ -49,13 +49,13 @@ WARP_TEMPLATES : WarpTemplateSet =\
[ 0.39032951, 0.77598822 ],
[ 0.61178945, 0.77476328 ]
]),
'styleganex_512': numpy.array(
'styleganex_384': numpy.array(
[
[ 0.43907768, 0.54098284 ],
[ 0.56204778, 0.54122359 ],
[ 0.42353745, 0.52289879 ],
[ 0.57725008, 0.52319972 ],
[ 0.50123859, 0.61331904 ],
[ 0.44716341, 0.66936502 ],
[ 0.55637032, 0.66911184 ]
[ 0.43364461, 0.68337652 ],
[ 0.57015325, 0.68306005 ]
])
}

View File

@ -32,28 +32,28 @@ MODEL_SET : ModelSet =\
{
'age_modifier':
{
'url': 'https://github.com/facefusion/facefusion-assets/releases/download/models-3.0.0/styleganex_age.hash',
'path': resolve_relative_path('../.assets/models/styleganex_age.hash')
'url': 'https://huggingface.co/bluefoxcreation/StyleGANEX-AGE/resolve/main/styleganex_age_opt.hash',
'path': resolve_relative_path('../.assets/models/styleganex_age_opt.hash')
}
},
'sources':
{
'age_modifier':
{
'url': 'https://github.com/facefusion/facefusion-assets/releases/download/models-3.0.0/styleganex_age.onnx',
'path': resolve_relative_path('../.assets/models/styleganex_age.onnx')
'url': 'https://huggingface.co/bluefoxcreation/StyleGANEX-AGE/resolve/main/styleganex_age_opt.onnx',
'path': resolve_relative_path('../.assets/models/styleganex_age_opt.onnx')
}
},
'templates':
{
'target': 'ffhq_512',
'target_with_background': 'styleganex_512'
'target_with_background': 'styleganex_384'
},
'sizes':
{
'target': (256, 256),
'target_with_background': (512, 512)
'target_with_background': (384, 384)
}
}
}

View File

@ -85,7 +85,7 @@ ProcessStep = Callable[[str, int, Args], bool]
Content = Dict[str, Any]
WarpTemplate = Literal['arcface_112_v1', 'arcface_112_v2', 'arcface_128_v2', 'ffhq_512', 'mtcnn_512', 'styleganex_512']
WarpTemplate = Literal['arcface_112_v1', 'arcface_112_v2', 'arcface_128_v2', 'ffhq_512', 'mtcnn_512', 'styleganex_384']
WarpTemplateSet = Dict[WarpTemplate, NDArray[Any]]
ProcessMode = Literal['output', 'preview', 'stream']