From 136cdcdf5b26a418c30daadf630acf08efa57735 Mon Sep 17 00:00:00 2001 From: adamnsandle Date: Fri, 28 Jun 2024 14:13:18 +0000 Subject: [PATCH 1/2] tst --- hubconf.py | 1 + 1 file changed, 1 insertion(+) diff --git a/hubconf.py b/hubconf.py index b4e8885..ab164b2 100644 --- a/hubconf.py +++ b/hubconf.py @@ -37,6 +37,7 @@ def silero_vad(onnx=False, force_onnx_cpu=False): raise Exception(f'Please install torch {supported_version} or greater ({installed_version} installed)') model_dir = os.path.join(os.path.dirname(__file__), 'files') + print(model_dir, os.path.dirname(__file__)) if onnx: model = OnnxWrapper(os.path.join(model_dir, 'silero_vad.onnx'), force_onnx_cpu) else: From 902cfc9248e102bb8ce2aec923cedb5000941e14 Mon Sep 17 00:00:00 2001 From: adamnsandle Date: Mon, 1 Jul 2024 09:00:59 +0000 Subject: [PATCH 2/2] fx dtype bug --- hubconf.py | 1 - utils_vad.py | 2 +- 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/hubconf.py b/hubconf.py index ab164b2..b4e8885 100644 --- a/hubconf.py +++ b/hubconf.py @@ -37,7 +37,6 @@ def silero_vad(onnx=False, force_onnx_cpu=False): raise Exception(f'Please install torch {supported_version} or greater ({installed_version} installed)') model_dir = os.path.join(os.path.dirname(__file__), 'files') - print(model_dir, os.path.dirname(__file__)) if onnx: model = OnnxWrapper(os.path.join(model_dir, 'silero_vad.onnx'), force_onnx_cpu) else: diff --git a/utils_vad.py b/utils_vad.py index 88c11b9..19be24c 100644 --- a/utils_vad.py +++ b/utils_vad.py @@ -72,7 +72,7 @@ class OnnxWrapper(): x = torch.cat([self._context, x], dim=1) if sr in [8000, 16000]: - ort_inputs = {'input': x.numpy(), 'state': self._state.numpy(), 'sr': np.array(sr)} + ort_inputs = {'input': x.numpy(), 'state': self._state.numpy(), 'sr': np.array(sr, dtype='int64')} ort_outs = self.session.run(None, ort_inputs) out, state = ort_outs self._state = torch.from_numpy(state)