diff --git a/facefusion/inference_manager.py b/facefusion/inference_manager.py index 53797b8c..8d5a3330 100644 --- a/facefusion/inference_manager.py +++ b/facefusion/inference_manager.py @@ -32,7 +32,7 @@ def get_inference_pool(module_name : str, model_names : List[str], model_source_ for execution_device_id in execution_device_ids: inference_context = get_inference_context(module_name, model_names, execution_device_id, execution_providers) - if state_manager.get_item('video_memory_strategy') in [ 'moderate', 'tolerant' ]: + if state_manager.get_item('video_memory_strategy') == 'tolerant': if app_context == 'cli' and INFERENCE_POOL_SET.get('ui').get(inference_context): INFERENCE_POOL_SET['cli'][inference_context] = INFERENCE_POOL_SET.get('ui').get(inference_context) if app_context == 'ui' and INFERENCE_POOL_SET.get('cli').get(inference_context):