Upload 50 files

modules/F0Predictor/RMVPEF0Predictor.py

@@ -16,7 +16,8 @@ class RMVPEF0Predictor(F0Predictor):
         self.f0_min = f0_min
         self.f0_max = f0_max
         if device is None:
-            self.device = 'cuda' if torch.cuda.is_available() else 'cpu'
+            self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+            #self.device = 'cuda' if torch.cuda.is_available() else 'cpu'
         else:
             self.device = device
         self.threshold = threshold
@@ -103,4 +104,4 @@ class RMVPEF0Predictor(F0Predictor):
         if torch.all(f0 == 0):
             rtn = f0.cpu().numpy() if p_len is None else np.zeros(p_len)
             return rtn,rtn
-        return self.post_process(x,self.sampling_rate,f0,p_len)
+        return self.post_process(x,self.sampling_rate,f0,p_len)