From 885a9b044f3b9c6718fc30526db4b9e3e873ba24 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Thu, 12 Jun 2025 17:43:54 +0200 Subject: [PATCH] Kill some elif and else --- facefusion/processors/modules/face_swapper.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/facefusion/processors/modules/face_swapper.py b/facefusion/processors/modules/face_swapper.py index 1df1fa0..859d1b6 100755 --- a/facefusion/processors/modules/face_swapper.py +++ b/facefusion/processors/modules/face_swapper.py @@ -508,14 +508,17 @@ def prepare_source_embedding(source_face : Face) -> Embedding: if model_type == 'ghost': source_embedding, _ = convert_embedding(source_face) source_embedding = source_embedding.reshape(1, -1) - elif model_type == 'inswapper': + return source_embedding + + if model_type == 'inswapper': model_path = get_model_options().get('sources').get('face_swapper').get('path') model_initializer = get_static_model_initializer(model_path) source_embedding = source_face.embedding.reshape((1, -1)) source_embedding = numpy.dot(source_embedding, model_initializer) / numpy.linalg.norm(source_embedding) - else: - _, source_normed_embedding = convert_embedding(source_face) - source_embedding = source_normed_embedding.reshape(1, -1) + return source_embedding + + _, source_normed_embedding = convert_embedding(source_face) + source_embedding = source_normed_embedding.reshape(1, -1) return source_embedding