Phospheneser commited on
Commit
aaea525
·
verified ·
1 Parent(s): 13ef6b2

fix: audio device for different torchaudio version

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -1912,7 +1912,7 @@ def extract_speech_token(model, feature_extractor, utts, batch_size=128):
1912
  # make sure kernel also on device
1913
  if hasattr(_resample_buffer[sample_rate], "kernel"):
1914
  _resample_buffer[sample_rate].kernel = _resample_buffer[sample_rate].kernel.to(device)
1915
- if torchaudio.__version__ == "2.8.0":
1916
  audio_device = audio.device
1917
  audio = audio.cpu()
1918
  audio = _resample_buffer[sample_rate](audio)
 
1912
  # make sure kernel also on device
1913
  if hasattr(_resample_buffer[sample_rate], "kernel"):
1914
  _resample_buffer[sample_rate].kernel = _resample_buffer[sample_rate].kernel.to(device)
1915
+ if "2.8.0" in torchaudio.__version__:
1916
  audio_device = audio.device
1917
  audio = audio.cpu()
1918
  audio = _resample_buffer[sample_rate](audio)