vad fix

2026-04-08 04:42:12 -04:00
parent c07187aae1
commit 08c5757b31
1 changed files with 3 additions and 5 deletions
@@ -18,8 +18,7 @@ class SileroVADOnnx:
        self._reset_state()

    def _reset_state(self):
-        self._h = np.zeros((2, 1, 64), dtype=np.float32)
-        self._c = np.zeros((2, 1, 64), dtype=np.float32)
+        self._state = np.zeros((2, 1, 128), dtype=np.float32)

    def reset_states(self):
        self._reset_state()
@@ -32,10 +31,9 @@ class SileroVADOnnx:
        ort_inputs = {
            "input": input_data,
            "sr": sr,
-            "h": self._h,
-            "c": self._c,
+            "state": self._state,
        }
-        out, self._h, self._c = self.session.run(None, ort_inputs)
+        out, self._state = self.session.run(None, ort_inputs)
        return float(out.squeeze())