Lucas Hansen commited on
Commit
96b3c52
1 Parent(s): 500769f

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

Browse files
facefusion/processors/frame/modules/face_enhancer.py CHANGED
@@ -82,7 +82,7 @@ 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 = [ 'CPUExecutionProvider' ])
86
  # FRAME_PROCESSOR = onnxruntime.InferenceSession(model_path, providers = facefusion.globals.execution_providers)
87
  return FRAME_PROCESSOR
88
 
 
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
  return FRAME_PROCESSOR
88