diff --git a/facefusion/face_detector.py b/facefusion/face_detector.py index c6345f71..09d104b2 100644 --- a/facefusion/face_detector.py +++ b/facefusion/face_detector.py @@ -75,14 +75,12 @@ MODEL_SET : ModelSet =\ def get_inference_pool() -> InferencePool: _, model_sources = collect_model_downloads() - face_detector_model = state_manager.get_item('face_detector_model') - model_context = __name__ + '.' + face_detector_model + model_context = __name__ + '.' + state_manager.get_item('face_detector_model') return inference_manager.get_inference_pool(model_context, model_sources) def clear_inference_pool() -> None: - face_detector_model = state_manager.get_item('face_detector_model') - model_context = __name__ + '.' + face_detector_model + model_context = __name__ + '.' + state_manager.get_item('face_detector_model') inference_manager.clear_inference_pool(model_context) diff --git a/facefusion/face_landmarker.py b/facefusion/face_landmarker.py index 15949cf5..a495404d 100644 --- a/facefusion/face_landmarker.py +++ b/facefusion/face_landmarker.py @@ -76,14 +76,12 @@ MODEL_SET : ModelSet =\ def get_inference_pool() -> InferencePool: _, model_sources = collect_model_downloads() - face_landmarker_model = state_manager.get_item('face_landmarker_model') - model_context = __name__ + '.' + face_landmarker_model + model_context = __name__ + '.' + state_manager.get_item('face_landmarker_model') return inference_manager.get_inference_pool(model_context, model_sources) def clear_inference_pool() -> None: - face_landmarker_model = state_manager.get_item('face_landmarker_model') - model_context = __name__ + '.' + face_landmarker_model + model_context = __name__ + '.' + state_manager.get_item('face_landmarker_model') inference_manager.clear_inference_pool(model_context) diff --git a/facefusion/processors/modules/age_modifier.py b/facefusion/processors/modules/age_modifier.py index 1792c44a..c1e39498 100755 --- a/facefusion/processors/modules/age_modifier.py +++ b/facefusion/processors/modules/age_modifier.py @@ -53,14 +53,12 @@ MODEL_SET : ModelSet =\ def get_inference_pool() -> InferencePool: model_sources = get_model_options().get('sources') - age_modifier_model = state_manager.get_item('age_modifier_model') - model_context = __name__ + '.' + age_modifier_model + model_context = __name__ + '.' + state_manager.get_item('age_modifier_model') return inference_manager.get_inference_pool(model_context, model_sources) def clear_inference_pool() -> None: - age_modifier_model = state_manager.get_item('age_modifier_model') - model_context = __name__ + '.' + age_modifier_model + model_context = __name__ + '.' + state_manager.get_item('age_modifier_model') inference_manager.clear_inference_pool(model_context) diff --git a/facefusion/processors/modules/expression_restorer.py b/facefusion/processors/modules/expression_restorer.py index 3fdffe03..d813aa37 100755 --- a/facefusion/processors/modules/expression_restorer.py +++ b/facefusion/processors/modules/expression_restorer.py @@ -73,8 +73,7 @@ MODEL_SET : ModelSet =\ def get_inference_pool() -> InferencePool: model_sources = get_model_options().get('sources') - expression_restorer_model = state_manager.get_item('expression_restorer_model') - model_context = __name__ + '.' + expression_restorer_model + model_context = __name__ + '.' + state_manager.get_item('expression_restorer_model') return inference_manager.get_inference_pool(model_context, model_sources) diff --git a/facefusion/processors/modules/face_editor.py b/facefusion/processors/modules/face_editor.py index b2d3c58a..cfd702b3 100755 --- a/facefusion/processors/modules/face_editor.py +++ b/facefusion/processors/modules/face_editor.py @@ -92,14 +92,12 @@ MODEL_SET : ModelSet =\ def get_inference_pool() -> InferencePool: model_sources = get_model_options().get('sources') - face_editor_model = state_manager.get_item('face_editor_model') - model_context = __name__ + '.' + face_editor_model + model_context = __name__ + '.' + state_manager.get_item('face_editor_model') return inference_manager.get_inference_pool(model_context, model_sources) def clear_inference_pool() -> None: - face_editor_model = state_manager.get_item('face_editor_model') - model_context = __name__ + '.' + face_editor_model + model_context = __name__ + '.' + state_manager.get_item('face_editor_model') inference_manager.clear_inference_pool(model_context) diff --git a/facefusion/processors/modules/face_enhancer.py b/facefusion/processors/modules/face_enhancer.py index a0c2acdc..da8d6498 100755 --- a/facefusion/processors/modules/face_enhancer.py +++ b/facefusion/processors/modules/face_enhancer.py @@ -219,14 +219,12 @@ MODEL_SET : ModelSet =\ def get_inference_pool() -> InferencePool: model_sources = get_model_options().get('sources') - face_enhancer_model = state_manager.get_item('face_enhancer_model') - model_context = __name__ + '.' + face_enhancer_model + model_context = __name__ + '.' + state_manager.get_item('face_enhancer_model') return inference_manager.get_inference_pool(model_context, model_sources) def clear_inference_pool() -> None: - face_enhancer_model = state_manager.get_item('face_enhancer_model') - model_context = __name__ + '.' + face_enhancer_model + model_context = __name__ + '.' + state_manager.get_item('face_enhancer_model') inference_manager.clear_inference_pool(model_context) diff --git a/facefusion/processors/modules/face_swapper.py b/facefusion/processors/modules/face_swapper.py index a7358d2b..276f8e5e 100755 --- a/facefusion/processors/modules/face_swapper.py +++ b/facefusion/processors/modules/face_swapper.py @@ -338,14 +338,12 @@ MODEL_SET : ModelSet =\ def get_inference_pool() -> InferencePool: model_sources = get_model_options().get('sources') - face_swapper_model = state_manager.get_item('face_swapper_model') - model_context = __name__ + '.' + face_swapper_model + model_context = __name__ + '.' + state_manager.get_item('face_swapper_model') return inference_manager.get_inference_pool(model_context, model_sources) def clear_inference_pool() -> None: - face_swapper_model = state_manager.get_item('face_swapper_model') - model_context = __name__ + '.' + face_swapper_model + model_context = __name__ + '.' + state_manager.get_item('face_swapper_model') inference_manager.clear_inference_pool(model_context) diff --git a/facefusion/processors/modules/frame_colorizer.py b/facefusion/processors/modules/frame_colorizer.py index 04f65e90..43f6b3d7 100644 --- a/facefusion/processors/modules/frame_colorizer.py +++ b/facefusion/processors/modules/frame_colorizer.py @@ -125,14 +125,12 @@ MODEL_SET : ModelSet =\ def get_inference_pool() -> InferencePool: model_sources = get_model_options().get('sources') - frame_colorizer_model = state_manager.get_item('frame_colorizer_model') - model_context = __name__ + '.' + frame_colorizer_model + model_context = __name__ + '.' + state_manager.get_item('frame_colorizer_model') return inference_manager.get_inference_pool(model_context, model_sources) def clear_inference_pool() -> None: - frame_colorizer_model = state_manager.get_item('frame_colorizer_model') - model_context = __name__ + '.' + frame_colorizer_model + model_context = __name__ + '.' + state_manager.get_item('frame_colorizer_model') inference_manager.clear_inference_pool(model_context) diff --git a/facefusion/processors/modules/frame_enhancer.py b/facefusion/processors/modules/frame_enhancer.py index d97adb14..8e9c9df2 100644 --- a/facefusion/processors/modules/frame_enhancer.py +++ b/facefusion/processors/modules/frame_enhancer.py @@ -277,14 +277,12 @@ MODEL_SET : ModelSet =\ def get_inference_pool() -> InferencePool: model_sources = get_model_options().get('sources') - frame_enhancer_model = state_manager.get_item('frame_enhancer_model') - model_context = __name__ + '.' + frame_enhancer_model + model_context = __name__ + '.' + state_manager.get_item('frame_enhancer_model') return inference_manager.get_inference_pool(model_context, model_sources) def clear_inference_pool() -> None: - frame_enhancer_model = state_manager.get_item('frame_enhancer_model') - model_context = __name__ + '.' + frame_enhancer_model + model_context = __name__ + '.' + state_manager.get_item('frame_enhancer_model') inference_manager.clear_inference_pool(model_context) diff --git a/facefusion/processors/modules/lip_syncer.py b/facefusion/processors/modules/lip_syncer.py index 5edf56d0..53f2a380 100755 --- a/facefusion/processors/modules/lip_syncer.py +++ b/facefusion/processors/modules/lip_syncer.py @@ -71,14 +71,12 @@ MODEL_SET : ModelSet =\ def get_inference_pool() -> InferencePool: model_sources = get_model_options().get('sources') - lip_syncer_model = state_manager.get_item('lip_syncer_model') - model_context = __name__ + '.' + lip_syncer_model + model_context = __name__ + '.' + state_manager.get_item('lip_syncer_model') return inference_manager.get_inference_pool(model_context, model_sources) def clear_inference_pool() -> None: - lip_syncer_model = state_manager.get_item('lip_syncer_model') - model_context = __name__ + '.' + lip_syncer_model + model_context = __name__ + '.' + state_manager.get_item('lip_syncer_model') inference_manager.clear_inference_pool(model_context)