fix(turn-detector): fix int32/64 errors on Windows (#1285)

livekit · Dec 23, 2024 · 37bbfcc · 37bbfcc
1 parent 799a53d
commit 37bbfcc
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/.changeset/tricky-spiders-change.md b/.changeset/tricky-spiders-change.md
@@ -0,0 +1,5 @@
+---
+"livekit-plugins-turn-detector": patch
+---
+
+fix int32/64 errors 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
@@ -106,7 +106,7 @@ def run(self, data: bytes) -> bytes | None:
             return_tensors="np",
         )
 
-        input_dict = {"input_ids": inputs["input_ids"]}
+        input_dict = {"input_ids": np.array(inputs["input_ids"], dtype=np.int64)}
 
         # Run inference
         outputs = self._session.run(["logits"], input_dict)