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
|