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

Kill some elif and else

Browse Source
This commit is contained in:
henryruhs
2025-06-12 17:43:54 +02:00
parent 4150cda64e
commit 885a9b044f
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -508,12 +508,15 @@ def prepare_source_embedding(source_face : Face) -> Embedding:
if model_type == 'ghost': if model_type == 'ghost':
source_embedding, _ = convert_embedding(source_face) source_embedding, _ = convert_embedding(source_face)
source_embedding = source_embedding.reshape(1, -1) 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_path = get_model_options().get('sources').get('face_swapper').get('path')
model_initializer = get_static_model_initializer(model_path) model_initializer = get_static_model_initializer(model_path)
source_embedding = source_face.embedding.reshape((1, -1)) source_embedding = source_face.embedding.reshape((1, -1))
source_embedding = numpy.dot(source_embedding, model_initializer) / numpy.linalg.norm(source_embedding) source_embedding = numpy.dot(source_embedding, model_initializer) / numpy.linalg.norm(source_embedding)
else: return source_embedding
_, source_normed_embedding = convert_embedding(source_face) _, source_normed_embedding = convert_embedding(source_face)
source_embedding = source_normed_embedding.reshape(1, -1) source_embedding = source_normed_embedding.reshape(1, -1)
return source_embedding return source_embedding