Spaces:
Running
Running
Lucas Hansen
commited on
Commit
•
f5476af
1
Parent(s):
fc6660c
Update facefusion/processors/frame/modules/face_enhancer.py
Browse files
facefusion/processors/frame/modules/face_enhancer.py
CHANGED
@@ -83,9 +83,10 @@ 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 |
-
|
88 |
-
|
|
|
89 |
return FRAME_PROCESSOR
|
90 |
|
91 |
|
|
|
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' ])
|
90 |
return FRAME_PROCESSOR
|
91 |
|
92 |
|