Spaces:
Running
Running
darksakura
commited on
Commit
•
93d7220
1
Parent(s):
4a46db4
Upload 50 files
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.
|
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(device)
|
23 |
self.threshold = threshold
|
24 |
self.sampling_rate = sampling_rate
|
25 |
self.dtype = dtype
|