ollieollie commited on
Commit
e7c60ec
·
verified ·
1 Parent(s): 05ec0bf

Update orator/src/orator/models/voice_encoder/voice_encoder.py

Browse files
orator/src/orator/models/voice_encoder/voice_encoder.py CHANGED
@@ -269,8 +269,6 @@ class VoiceEncoder(nn.Module):
269
  if "rate" not in kwargs:
270
  kwargs["rate"] = 1.3 # Resemble's default value.
271
 
272
- mel_func = melspectrogram()
273
- mels = [mel_func(torch.from_numpy(w)
274
- [None])[0].T for w in wavs]
275
 
276
  return self.embeds_from_mels(mels, as_spk=as_spk, batch_size=batch_size, **kwargs)
 
269
  if "rate" not in kwargs:
270
  kwargs["rate"] = 1.3 # Resemble's default value.
271
 
272
+ mels = [melspectrogram(w, self.hp).T for w in wavs]
 
 
273
 
274
  return self.embeds_from_mels(mels, as_spk=as_spk, batch_size=batch_size, **kwargs)