sindhuhegde commited on
Commit
ea0a289
·
1 Parent(s): 5d480ab

Update app

Browse files
Files changed (1) hide show
  1. utils/audio_utils.py +2 -2
utils/audio_utils.py CHANGED
@@ -35,8 +35,8 @@ def wav2filterbanks(wav, mel_basis=None):
35
  """
36
 
37
  assert len(wav.shape) == 2, 'Need batch of wavs as input'
38
- device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
39
- # device = 'cpu'
40
  spect = torch.stft(wav,
41
  return_complex=True,
42
  n_fft=audio_opts['n_fft'],
 
35
  """
36
 
37
  assert len(wav.shape) == 2, 'Need batch of wavs as input'
38
+ # device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
39
+ device = 'cpu'
40
  spect = torch.stft(wav,
41
  return_complex=True,
42
  n_fft=audio_opts['n_fft'],