Spaces:
Running
Running
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 = [ '
|
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 |
|