fix: onnxruntime-gpu not available on arm

speaches-ai · Jan 26, 2025 · bfab415 · bfab415
1 parent f27ec8b
commit bfab415
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -147,6 +147,8 @@ markers = [
 
 [tool.uv]
 constraint-dependencies = ["llvmlite>=0.40.0"]
+# TODO: create an issue on https://github.com/thewh1teagle/kokoro-onnx regarding this
+override-dependencies = ["onnxruntime-gpu>=1.20.1 ; platform_machine == 'x86_64'"]
 
 [tool.uv.sources]
 piper-phonemize = { url = "https://github.com/fedirz/piper-phonemize/raw/refs/heads/master/dist/piper_phonemize-1.2.0-cp312-cp312-manylinux_2_28_x86_64.whl" }

diff --git a/uv.lock b/uv.lock