Spaces:
Running
Running
Lucas Hansen
commited on
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' ])
|