Lucas Hansen commited on
Commit
6b58a69
1 Parent(s): 1ffebda

Update facefusion/processors/frame/modules/face_enhancer.py

Browse files
facefusion/processors/frame/modules/face_enhancer.py CHANGED
@@ -82,10 +82,10 @@ def get_frame_processor() -> Any:
82
  with THREAD_LOCK:
83
  if FRAME_PROCESSOR is None:
84
  model_path = get_options('model').get('path')
85
- FRAME_PROCESSOR = onnxruntime.InferenceSession(model_path, providers = [ 'CUDAExecutionProvider' ])
86
- # FRAME_PROCESSOR = onnxruntime.InferenceSession(model_path, providers = facefusion.globals.execution_providers)
87
- model_path = get_options('model').get('path')
88
- FRAME_PROCESSOR = onnxruntime.InferenceSession(model_path, providers = [ 'CPUExecutionProvider' ])
89
  return FRAME_PROCESSOR
90
 
91
 
 
82
  with THREAD_LOCK:
83
  if FRAME_PROCESSOR is None:
84
  model_path = get_options('model').get('path')
85
+ # FRAME_PROCESSOR = onnxruntime.InferenceSession(model_path, providers = [ 'CUDAExecutionProvider' ])
86
+ FRAME_PROCESSOR = onnxruntime.InferenceSession(model_path, providers = facefusion.globals.execution_providers)
87
+ # model_path = get_options('model').get('path')
88
+ # FRAME_PROCESSOR = onnxruntime.InferenceSession(model_path, providers = [ 'CPUExecutionProvider' ])
89
  return FRAME_PROCESSOR
90
 
91