tsi-org commited on
Commit
161514d
·
1 Parent(s): 2e7b2a8

Update roop/core.py

Browse files
Files changed (1) hide show
  1. 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