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