Lucas Hansen commited on
Commit
f5476af
1 Parent(s): fc6660c

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

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