Update uvr5/multiprocess_cuda_infer.py

uvr5/multiprocess_cuda_infer.py

@@ -222,9 +222,9 @@ class Inference():
         return x.reshape([-1,2,self.chunk_size])
 
 
-    def load_model(self, model_path, threads):
+    def load_model(self, model_path, threads, device='cpu'):
         model = onnx.load_model(model_path)
-        if torch.cuda.is_available():
+        if torch.cuda.is_available() and device != 'cpu':
             providers = [("CUDAExecutionProvider", {"device_id": torch.cuda.current_device(),
                                                     "user_compute_stream": str(torch.cuda.current_stream().cuda_stream)})]
         else: