diff --git a/pyproject.toml b/pyproject.toml index 329dbfb6..a2101746 100644 --- 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 index e9102d14..ea5044fb 100644 --- a/uv.lock +++ b/uv.lock @@ -3,6 +3,7 @@ requires-python = "==3.12.*" [manifest] constraints = [{ name = "llvmlite", specifier = ">=0.40.0" }] +overrides = [{ name = "onnxruntime-gpu", marker = "platform_machine == 'x86_64'", specifier = ">=1.20.1" }] [[manifest.dependency-metadata]] name = "piper-tts" @@ -878,7 +879,7 @@ wheels = [ [package.optional-dependencies] gpu = [ - { name = "onnxruntime-gpu", marker = "sys_platform != 'darwin'" }, + { name = "onnxruntime-gpu", marker = "platform_machine == 'x86_64'" }, ] [[package]]