From 40bf4ad4da5c69f5ba39acb3faff898ece95afdb Mon Sep 17 00:00:00 2001 From: Theo Date: Tue, 27 Jan 2026 21:43:31 +0000 Subject: [PATCH] torch fall back iwth runtime errors --- wsds/ws_audio.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/wsds/ws_audio.py b/wsds/ws_audio.py index ce1caa8..3720040 100644 --- a/wsds/ws_audio.py +++ b/wsds/ws_audio.py @@ -90,7 +90,7 @@ def _audio_to_mp3(samples): try: from torchcodec.encoders import AudioEncoder AudioEncoder(samples, sample_rate=int(samples.sample_rate)).to_file_like(out, "mp3") - except ImportError: + except (ImportError, RuntimeError): import torchaudio torchaudio.save(out, samples, int(samples.sample_rate), format="mp3") @@ -127,7 +127,7 @@ def get_reader(self, sample_rate=None): if self.reader is None or sample_rate_switch: try: from torchcodec.decoders import AudioDecoder - except ImportError: + except (ImportError, RuntimeError): AudioDecoder = CompatAudioDecoder reader = AudioDecoder(to_filelike(self.src), sample_rate=sample_rate)