Fix update of lip syncer and expression restorer

This commit is contained in:
henryruhs 2024-09-16 19:28:23 +02:00
parent a4dd8c95df
commit cbb32b4204
2 changed files with 2 additions and 2 deletions

View File

@ -51,7 +51,7 @@ def remote_update(processors : List[str]) -> Tuple[gradio.Dropdown, gradio.Slide
def update_expression_restorer_model(expression_restorer_model : ExpressionRestorerModel) -> gradio.Dropdown:
expression_restorer_module = load_processor_module('expression_restorer')
expression_restorer_module.clear_processor()
expression_restorer_module.clear_inference_pool()
state_manager.set_item('expression_restorer_model', expression_restorer_model)
if expression_restorer_module.pre_check():

View File

@ -38,7 +38,7 @@ def remote_update(processors : List[str]) -> gradio.Dropdown:
def update_lip_syncer_model(lip_syncer_model : LipSyncerModel) -> gradio.Dropdown:
lip_syncer_module = load_processor_module('lip_syncer')
lip_syncer_module.clear_processor()
lip_syncer_module.clear_inference_pool()
state_manager.set_item('lip_syncer_model', lip_syncer_model)
if lip_syncer_module.pre_check():