From 505bd9c0b522674e3782f3393644e3f2d7d238ba Mon Sep 17 00:00:00 2001 From: Barabazs <31799121+Barabazs@users.noreply.github.com> Date: Sun, 12 Oct 2025 02:38:29 -0600 Subject: [PATCH] chore: refine triton dependency to restrict installation to x86_64 Linux (#1259) * chore: refine triton dependency to restrict installation to x86_64 Linux * bump: update version to 3.7.2 --- pyproject.toml | 4 ++-- uv.lock | 18 ++++++------------ 2 files changed, 8 insertions(+), 14 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index fb78d4b..bdb3c54 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -2,7 +2,7 @@ urls = { repository = "https://github.com/m-bain/whisperx" } authors = [{ name = "Max Bain" }] name = "whisperx" -version = "3.7.1" +version = "3.7.2" description = "Time-Accurate Automatic Speech Recognition using Whisper." readme = "README.md" requires-python = ">=3.9, <3.14" @@ -22,7 +22,7 @@ dependencies = [ "torch>=2.7.1", "torchaudio", "transformers>=4.48.0", - "triton>=3.3.0; sys_platform == 'linux'" # only install triton on Linux + "triton>=3.3.0; sys_platform == 'linux' and platform_machine == 'x86_64'" # only install triton on x86_64 Linux ] diff --git a/uv.lock b/uv.lock index a86b1f9..07f1daa 100644 --- a/uv.lock +++ b/uv.lock @@ -1102,7 +1102,7 @@ name = "importlib-metadata" version = "8.7.0" source = { registry = "https://pypi.org/simple" } dependencies = [ - { name = "zipp", marker = "python_full_version < '3.10' and sys_platform != 'darwin'" }, + { name = "zipp", marker = "python_full_version < '3.10' and platform_machine == 'x86_64' and sys_platform != 'darwin'" }, ] sdist = { url = "https://files.pythonhosted.org/packages/76/66/650a33bd90f786193e4de4b3ad86ea60b53c89b669a5c7be931fac31cdb0/importlib_metadata-8.7.0.tar.gz", hash = "sha256:d13b81ad223b890aa16c5471f2ac3056cf76c5f10f82d6f9292f0b415f389000", size = 56641, upload-time = "2025-04-27T15:29:01.736Z" } wheels = [ @@ -3949,21 +3949,15 @@ name = "triton" version = "3.4.0" source = { registry = "https://download.pytorch.org/whl/cu128" } dependencies = [ - { name = "importlib-metadata", marker = "python_full_version < '3.10' and sys_platform != 'darwin'" }, - { name = "setuptools", marker = "sys_platform != 'darwin'" }, + { name = "importlib-metadata", marker = "python_full_version < '3.10' and platform_machine == 'x86_64' and sys_platform != 'darwin'" }, + { name = "setuptools", marker = "platform_machine == 'x86_64' and sys_platform != 'darwin'" }, ] wheels = [ - { url = "https://download.pytorch.org/whl/triton-3.4.0-cp310-cp310-manylinux_2_27_aarch64.manylinux_2_28_aarch64.whl" }, { url = "https://download.pytorch.org/whl/triton-3.4.0-cp310-cp310-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl" }, - { url = "https://download.pytorch.org/whl/triton-3.4.0-cp311-cp311-manylinux_2_27_aarch64.manylinux_2_28_aarch64.whl" }, { url = "https://download.pytorch.org/whl/triton-3.4.0-cp311-cp311-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl" }, - { url = "https://download.pytorch.org/whl/triton-3.4.0-cp312-cp312-manylinux_2_27_aarch64.manylinux_2_28_aarch64.whl" }, { url = "https://download.pytorch.org/whl/triton-3.4.0-cp312-cp312-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl" }, - { url = "https://download.pytorch.org/whl/triton-3.4.0-cp313-cp313-manylinux_2_27_aarch64.manylinux_2_28_aarch64.whl" }, { url = "https://download.pytorch.org/whl/triton-3.4.0-cp313-cp313-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl" }, - { url = "https://download.pytorch.org/whl/triton-3.4.0-cp313-cp313t-manylinux_2_27_aarch64.manylinux_2_28_aarch64.whl" }, { url = "https://download.pytorch.org/whl/triton-3.4.0-cp313-cp313t-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl" }, - { url = "https://download.pytorch.org/whl/triton-3.4.0-cp39-cp39-manylinux_2_27_aarch64.manylinux_2_28_aarch64.whl" }, { url = "https://download.pytorch.org/whl/triton-3.4.0-cp39-cp39-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl" }, ] @@ -4012,7 +4006,7 @@ wheels = [ [[package]] name = "whisperx" -version = "3.7.1" +version = "3.7.2" source = { editable = "." } dependencies = [ { name = "av" }, @@ -4031,7 +4025,7 @@ dependencies = [ { name = "torchaudio", version = "2.8.0+cpu", source = { registry = "https://download.pytorch.org/whl/cpu" }, marker = "(platform_machine != 'aarch64' and platform_machine != 'x86_64' and sys_platform == 'linux') or (platform_machine == 'aarch64' and platform_python_implementation != 'CPython' and sys_platform == 'linux') or (platform_machine != 'x86_64' and sys_platform != 'darwin' and sys_platform != 'linux')" }, { name = "torchaudio", version = "2.8.0+cu128", source = { registry = "https://download.pytorch.org/whl/cu128" }, marker = "platform_machine == 'x86_64' and sys_platform != 'darwin'" }, { name = "transformers" }, - { name = "triton", marker = "sys_platform == 'linux'" }, + { name = "triton", marker = "platform_machine == 'x86_64' and sys_platform == 'linux'" }, ] [package.metadata] @@ -4052,7 +4046,7 @@ requires-dist = [ { name = "torchaudio", marker = "platform_machine != 'x86_64' and sys_platform != 'darwin'", index = "https://download.pytorch.org/whl/cpu" }, { name = "torchaudio", marker = "platform_machine == 'x86_64' and sys_platform != 'darwin'", index = "https://download.pytorch.org/whl/cu128" }, { name = "transformers", specifier = ">=4.48.0" }, - { name = "triton", marker = "sys_platform == 'linux'", specifier = ">=3.3.0", index = "https://download.pytorch.org/whl/cu128" }, + { name = "triton", marker = "platform_machine == 'x86_64' and sys_platform == 'linux'", specifier = ">=3.3.0", index = "https://download.pytorch.org/whl/cu128" }, ] [[package]]