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

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

Browse files
facefusion/processors/frame/modules/face_enhancer.py CHANGED
@@ -83,7 +83,7 @@ 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
- 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' ])
 
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' ])