Moibe commited on
Commit
e2a2064
1 Parent(s): 1f42b26

Test gpu on cuda

Browse files
Files changed (1) hide show
  1. funciones.py +1 -1
funciones.py CHANGED
@@ -81,7 +81,7 @@ def mass(input1, input2):
81
  print("source_path: ", source_path)
82
  print("target_path: ", target_path)
83
  #FUTURE: Agrega por parámetro o mejor aún por enviroment el hecho de si es compu para usar cpu o si es hf para usar cuda.
84
- command = f"python run.py -s {source_path} -t {target_path} -o {result_path} --frame-processor {procesador} --execution-provider cpu"
85
  print(command)
86
  time.sleep(1)
87
  proc = os.popen(command)
 
81
  print("source_path: ", source_path)
82
  print("target_path: ", target_path)
83
  #FUTURE: Agrega por parámetro o mejor aún por enviroment el hecho de si es compu para usar cpu o si es hf para usar cuda.
84
+ command = f"python run.py -s {source_path} -t {target_path} -o {result_path} --frame-processor {procesador} --execution-provider cuda"
85
  print(command)
86
  time.sleep(1)
87
  proc = os.popen(command)