darksakura commited on
Commit
3fd8a60
1 Parent(s): ba49987

Upload RMVPEF0Predictor.py

Browse files
modules/F0Predictor/RMVPEF0Predictor.py CHANGED
@@ -19,7 +19,7 @@ class RMVPEF0Predictor(F0Predictor):
19
  self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
20
  #self.device = 'cuda' if torch.cuda.is_available() else 'cpu'
21
  else:
22
- self.dev = torch.device(device)
23
  self.threshold = threshold
24
  self.sampling_rate = sampling_rate
25
  self.dtype = dtype
 
19
  self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
20
  #self.device = 'cuda' if torch.cuda.is_available() else 'cpu'
21
  else:
22
+ self.dev = torch.device("cpu")
23
  self.threshold = threshold
24
  self.sampling_rate = sampling_rate
25
  self.dtype = dtype