This commit is contained in:
2026-04-08 04:42:12 -04:00
parent c07187aae1
commit 08c5757b31
+3 -5
View File
@@ -18,8 +18,7 @@ class SileroVADOnnx:
self._reset_state() self._reset_state()
def _reset_state(self): def _reset_state(self):
self._h = np.zeros((2, 1, 64), dtype=np.float32) self._state = np.zeros((2, 1, 128), dtype=np.float32)
self._c = np.zeros((2, 1, 64), dtype=np.float32)
def reset_states(self): def reset_states(self):
self._reset_state() self._reset_state()
@@ -32,10 +31,9 @@ class SileroVADOnnx:
ort_inputs = { ort_inputs = {
"input": input_data, "input": input_data,
"sr": sr, "sr": sr,
"h": self._h, "state": self._state,
"c": self._c,
} }
out, self._h, self._c = self.session.run(None, ort_inputs) out, self._state = self.session.run(None, ort_inputs)
return float(out.squeeze()) return float(out.squeeze())