Phospheneser commited on
Commit
d7c95ea
·
verified ·
1 Parent(s): 8f84585

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -4
utils.py CHANGED
@@ -1911,11 +1911,8 @@ def extract_speech_token(model, feature_extractor, utts, batch_size=128):
1911
  ).to(device)
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
-
1916
- audio = audio.cpu()
1917
  audio = _resample_buffer[sample_rate](audio)
1918
- audio = audio.to(device)
1919
  # if audio.shape[0] > 1:
1920
  # audio = audio[:1]
1921
  audio = audio[0]
 
1911
  ).to(device)
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
  audio = _resample_buffer[sample_rate](audio)
 
1916
  # if audio.shape[0] > 1:
1917
  # audio = audio[:1]
1918
  audio = audio[0]