From 41514c2d6731e51aa1b19686eff789ac4ccd7f2e Mon Sep 17 00:00:00 2001 From: henryruhs Date: Mon, 10 Feb 2025 19:30:32 +0100 Subject: [PATCH] Fix model size detection --- facefusion/processors/modules/deep_swapper.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/facefusion/processors/modules/deep_swapper.py b/facefusion/processors/modules/deep_swapper.py index 85d883f..6d9ed8f 100755 --- a/facefusion/processors/modules/deep_swapper.py +++ b/facefusion/processors/modules/deep_swapper.py @@ -258,10 +258,11 @@ def get_model_options() -> ModelOptions: def get_model_size() -> Size: deep_swapper = get_inference_pool().get('deep_swapper') - deep_swapper_outputs = deep_swapper.get_outputs() - for deep_swapper_output in deep_swapper_outputs: - return deep_swapper_output.shape[1:3] + for deep_swapper_input in deep_swapper.get_inputs(): + if deep_swapper_input.name == 'in_face:0': + return deep_swapper_input.shape[1:3] + return 0, 0