Spaces:
Build error
Build error
Update roop/core.py
Browse files- roop/core.py +5 -0
roop/core.py
CHANGED
|
@@ -202,6 +202,11 @@ def destroy() -> None:
|
|
| 202 |
|
| 203 |
|
| 204 |
def run() -> None:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 205 |
parse_args()
|
| 206 |
if not pre_check():
|
| 207 |
return
|
|
|
|
| 202 |
|
| 203 |
|
| 204 |
def run() -> None:
|
| 205 |
+
# Check if CUDA is available for ONNX Runtime and add to execution providers
|
| 206 |
+
if 'CUDAExecutionProvider' in onnxruntime.get_available_providers():
|
| 207 |
+
roop.globals.execution_providers.append('CUDAExecutionProvider')
|
| 208 |
+
# End of new lines
|
| 209 |
+
|
| 210 |
parse_args()
|
| 211 |
if not pre_check():
|
| 212 |
return
|