lord-reso commited on
Commit
2a96f72
·
verified ·
1 Parent(s): a2616e4

Update logic.py

Browse files
Files changed (1) hide show
  1. logic.py +1 -1
logic.py CHANGED
@@ -104,7 +104,7 @@ def extract_speech_embedding(audio_path: str):
104
  audio_norm = audio / 32768.0
105
  audio_norm = audio_norm.unsqueeze(0)
106
  audio_norm = torch.autograd.Variable(audio_norm, requires_grad=False).to(device)
107
- melspec = stft.mel_spectrogram(audio_norm).transpose(1,2).half()
108
 
109
  if melspec.shape[1] <= 128:
110
  mel_slice = mel
 
104
  audio_norm = audio / 32768.0
105
  audio_norm = audio_norm.unsqueeze(0)
106
  audio_norm = torch.autograd.Variable(audio_norm, requires_grad=False).to(device)
107
+ melspec = stft.mel_spectrogram(audio_norm).transpose(1,2).float()
108
 
109
  if melspec.shape[1] <= 128:
110
  mel_slice = mel