Spaces:
Running
Running
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 |
-
|
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 |
|