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

Drop has_inference_model and solve issue on Gradio side

Browse Source
This commit is contained in:
henryruhs
2025-01-14 00:44:35 +01:00
parent b11cb07aea
commit ed8e25dbb2
5 changed files with 24 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -238,10 +238,6 @@ def create_static_model_set(download_scope : DownloadScope) -> ModelSet:
return model_set
def has_inference_model(model_name : str) -> bool:
return inference_manager.has_inference_model(__name__, model_name)
def get_inference_pool() -> InferencePool:
model_sources = get_model_options().get('sources')
return inference_manager.get_inference_pool(__name__, model_sources)
@@ -361,12 +357,11 @@ def forward(crop_vision_frame : VisionFrame, deep_swapper_morph : DeepSwapperMor
def has_morph_input() -> bool:
if has_inference_model('deep_swapper'):
deep_swapper = get_inference_pool().get('deep_swapper')
deep_swapper = get_inference_pool().get('deep_swapper')
for deep_swapper_input in deep_swapper.get_inputs():
if deep_swapper_input.name == 'morph_value:0':
return True
for deep_swapper_input in deep_swapper.get_inputs():
if deep_swapper_input.name == 'morph_value:0':
return True
return False

13
facefusion/processors/modules/face_enhancer.py
View File

@@ -221,10 +221,6 @@ def create_static_model_set(download_scope : DownloadScope) -> ModelSet:
}
def has_inference_model(model_name : str) -> bool:
return inference_manager.has_inference_model(__name__, model_name)
def get_inference_pool() -> InferencePool:
model_sources = get_model_options().get('sources')
return inference_manager.get_inference_pool(__name__, model_sources)
@@ -328,12 +324,11 @@ def forward(crop_vision_frame : VisionFrame, face_enhancer_weight : FaceEnhancer
def has_weight_input() -> bool:
if has_inference_model('face_enhancer'):
face_enhancer = get_inference_pool().get('face_enhancer')
face_enhancer = get_inference_pool().get('face_enhancer')
for deep_swapper_input in face_enhancer.get_inputs():
if deep_swapper_input.name == 'weight':
return True
for deep_swapper_input in face_enhancer.get_inputs():
if deep_swapper_input.name == 'weight':
return True
return False