asr

Running on Zero

kgout commited on Apr 30

Commit

fadf1ae

verified ·

1 Parent(s): 6307588

Update audiosr/pipeline.py

Files changed (1) hide show

audiosr/pipeline.py CHANGED Viewed

@@ -115,7 +115,7 @@ 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.device("cuda:0")
         elif torch.backends.mps.is_available():
             device = torch.device("mps")
         else:

 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.device("cuda")
         elif torch.backends.mps.is_available():
             device = torch.device("mps")
         else: