Skip to content

Commit 4a3be20

Browse files
authored
Update encoders.py
1 parent 2fa8887 commit 4a3be20

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

encodecmae/tasks/models/encodecmae/encoders.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@ def __init__(self, frozen=True, scale=1.0):
88
model = EncodecModel.encodec_model_24khz()
99
self.model = model.encoder
1010
self.hop_length = self.model.hop_length
11+
self.fs = 24000
1112
self.frozen = frozen
1213
if self.frozen:
1314
self.model.eval()
@@ -22,4 +23,4 @@ def forward(self, x):
2223
else:
2324
y = self.model(x)
2425
y = torch.permute(y,(0,2,1))*self.scale
25-
return y
26+
return y

0 commit comments

Comments
 (0)