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

Add deep swapper models by Jen

Browse Source
This commit is contained in:
henryruhs 2024-11-20 14:23:47 +01:00
parent ba874607d7
commit ab34dbb991
4 changed files with 22 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
facefusion/processors/choices.py
View File

@ -101,6 +101,13 @@ deep_swapper_models : List[DeepSwapperModel] =\
'iperov/thomas_holland_224',
'iperov/vin_diesel_224',
'iperov/vladimir_putin_224',
'jen/angelica_trae_288',
'jen/ella_freya_224',
'jen/emma_myers_320',
'jen/evie_pickerill_224',
'jen/kang_hyewon_320',
'jen/maddie_mead_224',
'jen/nicole_turnbull_288',
'mats/alica_schmidt_320',
'mats/ashley_alexiss_224',
'mats/billie_eilish_224',

7
facefusion/processors/modules/deep_swapper.py
View File

@ -122,6 +122,13 @@ def create_model_set() -> ModelSet:
('iperov', 'thomas_holland_224', (224, 224)),
('iperov', 'vin_diesel_224', (224, 224)),
('iperov', 'vladimir_putin_224', (224, 224)),
('jen', 'angelica_trae_288', (288, 288)),
('jen', 'ella_freya_224', (224, 224)),
('jen', 'emma_myers_320', (320, 320)),
('jen', 'evie_pickerill_224', (224, 224)),
('jen', 'kang_hyewon_320', (320, 320)),
('jen', 'maddie_mead_224', (224, 224)),
('jen', 'nicole_turnbull_288', (288, 288)),
('mats', 'alica_schmidt_320', (320, 320)),
('mats', 'ashley_alexiss_224', (224, 224)),
('mats', 'billie_eilish_224', (224, 224)),

7
facefusion/processors/typing.py
View File

@ -102,6 +102,13 @@ DeepSwapperModel = Literal\
'iperov/thomas_holland_224',
'iperov/vin_diesel_224',
'iperov/vladimir_putin_224',
'jen/angelica_trae_288',
'jen/ella_freya_224',
'jen/emma_myers_320',
'jen/evie_pickerill_224',
'jen/kang_hyewon_320',
'jen/maddie_mead_224',
'jen/nicole_turnbull_288',
'mats/alica_schmidt_320',
'mats/ashley_alexiss_224',
'mats/billie_eilish_224',

2
facefusion/uis/ui_helper.py
View File

@ -21,6 +21,6 @@ def convert_str_none(value : str) -> Optional[str]:
def suggest_output_path(output_directory_path : str, target_path : str) -> Optional[str]:
if is_image(target_path) or is_video(target_path):
_, target_extension = os.path.splitext(target_path)
output_name = hashlib.sha1(str(state_manager.get_state()).encode('utf-8')).hexdigest()[:8]
output_name = hashlib.sha1(str(state_manager.get_state()).encode()).hexdigest()[:8]
return os.path.join(output_directory_path, output_name + target_extension)
return None