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

Fix/remote inference pool lookups (#848)

Browse Source 
* Fix edge case when offline and inference session has no model, Prevent inference session creation

* Fix edge case when offline and inference session has no model, Prevent inference session creation

* Fix edge case when offline and inference session has no model, Prevent inference session creation

* Fix edge case when offline and inference session has no model, Prevent inference session creation
This commit is contained in:
Henry Ruhs
2025-01-07 13:44:01 +01:00
committed by henryruhs
parent 87350eb45f
commit c5bc7c50a5
4 changed files with 39 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -238,6 +238,10 @@ 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)
@@ -357,11 +361,13 @@ def forward(crop_vision_frame : VisionFrame, deep_swapper_morph : DeepSwapperMor
def has_morph_input() -> bool:
deep_swapper = get_inference_pool().get('deep_swapper')
if has_inference_model('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

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

@@ -221,6 +221,10 @@ 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)
@@ -324,11 +328,13 @@ def forward(crop_vision_frame : VisionFrame, face_enhancer_weight : FaceEnhancer
def has_weight_input() -> bool:
face_enhancer = get_inference_pool().get('face_enhancer')
if has_inference_model('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