ensure correct dtype on windows (#1452)

livekit · Feb 5, 2025 · c0e2db3 · c0e2db3
1 parent 84c2e90
commit c0e2db3
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/.changeset/hungry-bees-tie.md b/.changeset/hungry-bees-tie.md
@@ -0,0 +1,5 @@
+---
+"livekit-plugins-turn-detector": patch
+---
+
+fix incorrect dtype on windows
diff --git a/livekit-plugins/livekit-plugins-turn-detector/livekit/plugins/turn_detector/eou.py b/livekit-plugins/livekit-plugins-turn-detector/livekit/plugins/turn_detector/eou.py
@@ -102,7 +102,9 @@ def run(self, data: bytes) -> bytes | None:
             truncation=True,
         )
         # Run inference
-        outputs = self._session.run(None, {"input_ids": inputs["input_ids"]})
+        outputs = self._session.run(
+            None, {"input_ids": inputs["input_ids"].astype("int64")}
+        )
         eou_probability = outputs[0][0]
         end_time = time.perf_counter()