richermans commited on
Commit
ce45305
·
verified ·
1 Parent(s): bd8db95

Update feature_extraction_dasheng.py

Browse files
Files changed (1) hide show
  1. feature_extraction_dasheng.py +2 -1
feature_extraction_dasheng.py CHANGED
@@ -164,7 +164,8 @@ class DashengFeatureExtractor(SequenceFeatureExtractor):
164
  raise ValueError(
165
  "Input must be a numpy array, a list of numpy arrays, a PyTorch tensor, or a list of PyTorch tensor."
166
  )
167
-
 
168
  x = x.float()
169
  x = mel_spectrogram(x)
170
  x = amplitude_to_db(x)
 
164
  raise ValueError(
165
  "Input must be a numpy array, a list of numpy arrays, a PyTorch tensor, or a list of PyTorch tensor."
166
  )
167
+
168
+ mel_spectrogram = mel_spectrogram.to(x.device)
169
  x = x.float()
170
  x = mel_spectrogram(x)
171
  x = amplitude_to_db(x)