From 1b59f8ab73dd59c5c80433b0d5b0c97c246a0efd Mon Sep 17 00:00:00 2001 From: henryruhs Date: Tue, 10 Dec 2024 16:45:32 +0100 Subject: [PATCH] Local model support for deep swapper part3 --- facefusion/processors/choices.py | 8 +++++--- facefusion/processors/modules/deep_swapper.py | 8 ++++---- 2 files changed, 9 insertions(+), 7 deletions(-) diff --git a/facefusion/processors/choices.py b/facefusion/processors/choices.py index 125e60a7..60214bd6 100755 --- a/facefusion/processors/choices.py +++ b/facefusion/processors/choices.py @@ -157,9 +157,11 @@ deep_swapper_models : List[DeepSwapperModel] =\ model_files = list_directory(resolve_relative_path('../.assets/models/local')) -for model_file in model_files: - model_id = '/'.join([ 'local', model_file.get('name') ]) - deep_swapper_models.append(model_id) +if model_files: + + for model_file in model_files: + model_id = '/'.join([ 'local', model_file.get('name') ]) + deep_swapper_models.append(model_id) expression_restorer_models : List[ExpressionRestorerModel] = [ 'live_portrait' ] face_debugger_items : List[FaceDebuggerItem] = [ 'bounding-box', 'face-landmark-5', 'face-landmark-5/68', 'face-landmark-68', 'face-landmark-68/5', 'face-mask', 'face-detector-score', 'face-landmarker-score', 'age', 'gender', 'race' ] diff --git a/facefusion/processors/modules/deep_swapper.py b/facefusion/processors/modules/deep_swapper.py index 5c0c41f4..52c477eb 100755 --- a/facefusion/processors/modules/deep_swapper.py +++ b/facefusion/processors/modules/deep_swapper.py @@ -225,10 +225,10 @@ def create_static_model_set(download_scope : DownloadScope) -> ModelSet: { 'sources': { - 'deep_swapper': - { - 'path': resolve_relative_path(model_file.get('path')) - } + 'deep_swapper': + { + 'path': resolve_relative_path(model_file.get('path')) + } }, 'template': 'dfl_whole_face' }