Update audiosr/pipeline.py

audiosr/pipeline.py

@@ -118,8 +118,8 @@ def round_up_duration(duration):
 def build_model(ckpt_path=None, config=None, device=None, model_name="basic"):
     if device is None or device == "auto":
         if torch.cuda.is_available():
-            device = torch.Tensor([0]).cuda()
-            # device = torch.device("cuda:0")
+            # device = torch.Tensor([0]).cuda()
+            device = torch.device("cuda:0")
         elif torch.backends.mps.is_available():
             device = torch.device("mps")
         else: