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

Fix model size detection

Browse Source
This commit is contained in:
henryruhs
2025-02-10 19:30:32 +01:00
parent 977d8d91b5
commit 41514c2d67
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -258,10 +258,11 @@ def get_model_options() -> ModelOptions:
def get_model_size() -> Size: def get_model_size() -> Size:
deep_swapper = get_inference_pool().get('deep_swapper') deep_swapper = get_inference_pool().get('deep_swapper')
deep_swapper_outputs = deep_swapper.get_outputs()
for deep_swapper_output in deep_swapper_outputs: for deep_swapper_input in deep_swapper.get_inputs():
return deep_swapper_output.shape[1:3] if deep_swapper_input.name == 'in_face:0':
return deep_swapper_input.shape[1:3]
return 0, 0 return 0, 0