fix: audio device for different torchaudio version
Browse files
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
|
| 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)
|