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

Simplify model context

Browse Source
This commit is contained in:
henryruhs 2024-12-18 00:42:35 +01:00
parent cf642e4e9d
commit e38324e6f6
11 changed files with 21 additions and 42 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
facefusion/face_detector.py
View File

@ -79,13 +79,11 @@ def create_static_model_set(download_scope : DownloadScope) -> ModelSet:
def get_inference_pool() -> InferencePool:
_, model_sources = collect_model_downloads()
model_context = __name__ + '.' + state_manager.get_item('face_detector_model')
return inference_manager.get_inference_pool(model_context, model_sources)
return inference_manager.get_inference_pool(__name__, model_sources)
def clear_inference_pool() -> None:
model_context = __name__ + '.' + state_manager.get_item('face_detector_model')
inference_manager.clear_inference_pool(model_context)
inference_manager.clear_inference_pool(__name__)
def collect_model_downloads() -> Tuple[DownloadSet, DownloadSet]:

6
facefusion/face_landmarker.py
View File

@ -80,13 +80,11 @@ def create_static_model_set(download_scope : DownloadScope) -> ModelSet:
def get_inference_pool() -> InferencePool:
_, model_sources = collect_model_downloads()
model_context = __name__ + '.' + state_manager.get_item('face_landmarker_model')
return inference_manager.get_inference_pool(model_context, model_sources)
return inference_manager.get_inference_pool(__name__, model_sources)
def clear_inference_pool() -> None:
model_context = __name__ + '.' + state_manager.get_item('face_landmarker_model')
inference_manager.clear_inference_pool(model_context)
inference_manager.clear_inference_pool(__name__)
def collect_model_downloads() -> Tuple[DownloadSet, DownloadSet]:

6
facefusion/processors/modules/age_modifier.py
View File

@ -65,13 +65,11 @@ def create_static_model_set(download_scope : DownloadScope) -> ModelSet:
def get_inference_pool() -> InferencePool:
model_sources = get_model_options().get('sources')
model_context = __name__ + '.' + state_manager.get_item('age_modifier_model')
return inference_manager.get_inference_pool(model_context, model_sources)
return inference_manager.get_inference_pool(__name__, model_sources)
def clear_inference_pool() -> None:
model_context = __name__ + '.' + state_manager.get_item('age_modifier_model')
inference_manager.clear_inference_pool(model_context)
inference_manager.clear_inference_pool(__name__)
def get_model_options() -> ModelOptions:

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

@ -240,13 +240,11 @@ def create_static_model_set(download_scope : DownloadScope) -> ModelSet:
def get_inference_pool() -> InferencePool:
model_sources = get_model_options().get('sources')
model_context = __name__ + '.' + state_manager.get_item('deep_swapper_model')
return inference_manager.get_inference_pool(model_context, model_sources)
return inference_manager.get_inference_pool(__name__, model_sources)
def clear_inference_pool() -> None:
model_context = __name__ + '.' + state_manager.get_item('deep_swapper_model')
inference_manager.clear_inference_pool(model_context)
inference_manager.clear_inference_pool(__name__)
def get_model_options() -> ModelOptions:

3
facefusion/processors/modules/expression_restorer.py
View File

@ -77,8 +77,7 @@ def create_static_model_set(download_scope : DownloadScope) -> ModelSet:
def get_inference_pool() -> InferencePool:
model_sources = get_model_options().get('sources')
model_context = __name__ + '.' + state_manager.get_item('expression_restorer_model')
return inference_manager.get_inference_pool(model_context, model_sources)
return inference_manager.get_inference_pool(__name__, model_sources)
def clear_inference_pool() -> None:

6
facefusion/processors/modules/face_editor.py
View File

@ -106,13 +106,11 @@ def create_static_model_set(download_scope : DownloadScope) -> ModelSet:
def get_inference_pool() -> InferencePool:
model_sources = get_model_options().get('sources')
model_context = __name__ + '.' + state_manager.get_item('face_editor_model')
return inference_manager.get_inference_pool(model_context, model_sources)
return inference_manager.get_inference_pool(__name__, model_sources)
def clear_inference_pool() -> None:
model_context = __name__ + '.' + state_manager.get_item('face_editor_model')
inference_manager.clear_inference_pool(model_context)
inference_manager.clear_inference_pool(__name__)
def get_model_options() -> ModelOptions:

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

@ -223,13 +223,11 @@ def create_static_model_set(download_scope : DownloadScope) -> ModelSet:
def get_inference_pool() -> InferencePool:
model_sources = get_model_options().get('sources')
model_context = __name__ + '.' + state_manager.get_item('face_enhancer_model')
return inference_manager.get_inference_pool(model_context, model_sources)
return inference_manager.get_inference_pool(__name__, model_sources)
def clear_inference_pool() -> None:
model_context = __name__ + '.' + state_manager.get_item('face_enhancer_model')
inference_manager.clear_inference_pool(model_context)
inference_manager.clear_inference_pool(__name__)
def get_model_options() -> ModelOptions:

6
facefusion/processors/modules/face_swapper.py
View File

@ -337,13 +337,11 @@ def create_static_model_set(download_scope : DownloadScope) -> ModelSet:
def get_inference_pool() -> InferencePool:
model_sources = get_model_options().get('sources')
model_context = __name__ + '.' + state_manager.get_item('face_swapper_model')
return inference_manager.get_inference_pool(model_context, model_sources)
return inference_manager.get_inference_pool(__name__, model_sources)
def clear_inference_pool() -> None:
model_context = __name__ + '.' + state_manager.get_item('face_swapper_model')
inference_manager.clear_inference_pool(model_context)
inference_manager.clear_inference_pool(__name__)
def get_model_options() -> ModelOptions:

6
facefusion/processors/modules/frame_colorizer.py
View File

@ -129,13 +129,11 @@ def create_static_model_set(download_scope : DownloadScope) -> ModelSet:
def get_inference_pool() -> InferencePool:
model_sources = get_model_options().get('sources')
model_context = __name__ + '.' + state_manager.get_item('frame_colorizer_model')
return inference_manager.get_inference_pool(model_context, model_sources)
return inference_manager.get_inference_pool(__name__, model_sources)
def clear_inference_pool() -> None:
model_context = __name__ + '.' + state_manager.get_item('frame_colorizer_model')
inference_manager.clear_inference_pool(model_context)
inference_manager.clear_inference_pool(__name__)
def get_model_options() -> ModelOptions:

6
facefusion/processors/modules/frame_enhancer.py
View File

@ -386,13 +386,11 @@ def create_static_model_set(download_scope : DownloadScope) -> ModelSet:
def get_inference_pool() -> InferencePool:
model_sources = get_model_options().get('sources')
model_context = __name__ + '.' + state_manager.get_item('frame_enhancer_model')
return inference_manager.get_inference_pool(model_context, model_sources)
return inference_manager.get_inference_pool(__name__, model_sources)
def clear_inference_pool() -> None:
model_context = __name__ + '.' + state_manager.get_item('frame_enhancer_model')
inference_manager.clear_inference_pool(model_context)
inference_manager.clear_inference_pool(__name__)
def get_model_options() -> ModelOptions:

6
facefusion/processors/modules/lip_syncer.py
View File

@ -75,13 +75,11 @@ def create_static_model_set(download_scope : DownloadScope) -> ModelSet:
def get_inference_pool() -> InferencePool:
model_sources = get_model_options().get('sources')
model_context = __name__ + '.' + state_manager.get_item('lip_syncer_model')
return inference_manager.get_inference_pool(model_context, model_sources)
return inference_manager.get_inference_pool(__name__, model_sources)
def clear_inference_pool() -> None:
model_context = __name__ + '.' + state_manager.get_item('lip_syncer_model')
inference_manager.clear_inference_pool(model_context)
inference_manager.clear_inference_pool(__name__)
def get_model_options() -> ModelOptions: