Update utils.py
Browse files
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]
|