File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -19,7 +19,7 @@ def __init__(
19
19
mel_fmin : int = 0 ,
20
20
mel_fmax : int = None ,
21
21
clamp : float = 1e-5 ,
22
- device = torch .device ("cpu" ),
22
+ device = torch .device ("cpu" ),
23
23
):
24
24
super ().__init__ ()
25
25
if n_fft is None :
@@ -45,7 +45,7 @@ def __init__(
45
45
hop_length = hop_length ,
46
46
win_length = win_length ,
47
47
window = "hann" ,
48
- use_torch_stft = "privateuseone" not in str (device )
48
+ use_torch_stft = "privateuseone" not in str (device ),
49
49
).to (device )
50
50
51
51
def forward (
Original file line number Diff line number Diff line change @@ -14,7 +14,7 @@ def __init__(
14
14
hop_length = 512 ,
15
15
win_length : Optional [int ] = None ,
16
16
window = "hann" ,
17
- use_torch_stft = True ,
17
+ use_torch_stft = True ,
18
18
):
19
19
"""
20
20
This module implements an STFT using 1D convolution and 1D transpose convolutions.
You can’t perform that action at this time.
0 commit comments