Spaces:
Running
Running
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 |
-
|
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' ])
|