deepgeek/silero-vad
2
0
Fork 0
mirror of https://github.com/snakers4/silero-vad.git synced 2026-02-05 18:09:22 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #481 from snakers4/adamnsandle

Browse Source 
Adamnsandle
This commit is contained in:
Dimitrii Voronin
2024-07-01 12:02:50 +03:00
committed by GitHub
parent a3bdebed16 902cfc9248
commit 78958b6fb6
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
utils_vad.py
View File

@@ -72,7 +72,7 @@ class OnnxWrapper():
x = torch.cat([self._context, x], dim=1) x = torch.cat([self._context, x], dim=1)
if sr in [8000, 16000]: 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) ort_outs = self.session.run(None, ort_inputs)
out, state = ort_outs out, state = ort_outs
self._state = torch.from_numpy(state) self._state = torch.from_numpy(state)