mirrors/facefusion
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix suffix of swapper and lip syncer models

Browse Source
This commit is contained in:
henryruhs 2024-10-03 18:18:50 +02:00
parent 423f1b6c40
commit c1282a72cd
1 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
facefusion/processors/modules/lip_syncer.py
View File

@ -26,42 +26,42 @@ from facefusion.vision import read_image, read_static_image, restrict_video_fps,
MODEL_SET : ModelSet =\ MODEL_SET : ModelSet =\
{ {
'wav2lip': 'wav2lip_96':
{ {
'hashes': 'hashes':
{ {
'lip_syncer': 'lip_syncer':
{ {
'url': 'https://github.com/facefusion/facefusion-assets/releases/download/models-3.0.0/wav2lip.hash', 'url': 'https://github.com/facefusion/facefusion-assets/releases/download/models-3.0.0/wav2lip_96.hash',
'path': resolve_relative_path('../.assets/models/wav2lip.hash') 'path': resolve_relative_path('../.assets/models/wav2lip_96.hash')
} }
}, },
'sources': 'sources':
{ {
'lip_syncer': 'lip_syncer':
{ {
'url': 'https://github.com/facefusion/facefusion-assets/releases/download/models-3.0.0/wav2lip.onnx', 'url': 'https://github.com/facefusion/facefusion-assets/releases/download/models-3.0.0/wav2lip_96.onnx',
'path': resolve_relative_path('../.assets/models/wav2lip.onnx') 'path': resolve_relative_path('../.assets/models/wav2lip_96.onnx')
} }
}, },
'size': (96, 96) 'size': (96, 96)
}, },
'wav2lip_gan': 'wav2lip_gan_96':
{ {
'hashes': 'hashes':
{ {
'lip_syncer': 'lip_syncer':
{ {
'url': 'https://github.com/facefusion/facefusion-assets/releases/download/models-3.0.0/wav2lip_gan.hash', 'url': 'https://github.com/facefusion/facefusion-assets/releases/download/models-3.0.0/wav2lip_gan_96.hash',
'path': resolve_relative_path('../.assets/models/wav2lip_gan.hash') 'path': resolve_relative_path('../.assets/models/wav2lip_gan_96.hash')
} }
}, },
'sources': 'sources':
{ {
'lip_syncer': 'lip_syncer':
{ {
'url': 'https://github.com/facefusion/facefusion-assets/releases/download/models-3.0.0/wav2lip_gan.onnx', 'url': 'https://github.com/facefusion/facefusion-assets/releases/download/models-3.0.0/wav2lip_gan_96.onnx',
'path': resolve_relative_path('../.assets/models/wav2lip_gan.onnx') 'path': resolve_relative_path('../.assets/models/wav2lip_gan_96.onnx')
} }
}, },
'size': (96, 96) 'size': (96, 96)
@ -88,7 +88,7 @@ def get_model_options() -> ModelOptions:
def register_args(program : ArgumentParser) -> None: def register_args(program : ArgumentParser) -> None:
group_processors = find_argument_group(program, 'processors') group_processors = find_argument_group(program, 'processors')
if group_processors: if group_processors:
group_processors.add_argument('--lip-syncer-model', help = wording.get('help.lip_syncer_model'), default = config.get_str_value('processors.lip_syncer_model', 'wav2lip_gan'), choices = processors_choices.lip_syncer_models) group_processors.add_argument('--lip-syncer-model', help = wording.get('help.lip_syncer_model'), default = config.get_str_value('processors.lip_syncer_model', 'wav2lip_gan_96'), choices = processors_choices.lip_syncer_models)
facefusion.jobs.job_store.register_step_keys([ 'lip_syncer_model' ]) facefusion.jobs.job_store.register_step_keys([ 'lip_syncer_model' ])