Phospheneser commited on
Commit
d3c20ea
·
verified ·
1 Parent(s): f4a612c

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +4 -0
utils.py CHANGED
@@ -1909,6 +1909,10 @@ def extract_speech_token(model, feature_extractor, utts, batch_size=128):
1909
  orig_freq=sample_rate,
1910
  new_freq=16000
1911
  ).to(device)
 
 
 
 
1912
  audio = _resample_buffer[sample_rate](audio)
1913
  # if audio.shape[0] > 1:
1914
  # audio = audio[:1]
 
1909
  orig_freq=sample_rate,
1910
  new_freq=16000
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 = _resample_buffer[sample_rate](audio)
1917
  # if audio.shape[0] > 1:
1918
  # audio = audio[:1]