vlad.os/whisperx-rocm-api
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore: refine triton dependency to restrict installation to x86_64 Linux (#1259)

Browse Source 
* chore: refine triton dependency to restrict installation to x86_64 Linux

* bump: update version to 3.7.2
This commit is contained in:
Barabazs
2025-10-12 02:38:29 -06:00
committed by GitHub
parent 895e5a8493
commit 505bd9c0b5
2 changed files with 8 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
pyproject.toml
View File

@@ -2,7 +2,7 @@
urls = { repository = "https://github.com/m-bain/whisperx" } urls = { repository = "https://github.com/m-bain/whisperx" }
authors = [{ name = "Max Bain" }] authors = [{ name = "Max Bain" }]
name = "whisperx" name = "whisperx"
version = "3.7.1" version = "3.7.2"
description = "Time-Accurate Automatic Speech Recognition using Whisper." description = "Time-Accurate Automatic Speech Recognition using Whisper."
readme = "README.md" readme = "README.md"
requires-python = ">=3.9, <3.14" requires-python = ">=3.9, <3.14"
@@ -22,7 +22,7 @@ dependencies = [
"torch>=2.7.1", "torch>=2.7.1",
"torchaudio", "torchaudio",
"transformers>=4.48.0", "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
] ]

18
uv.lock generated
View File

@@ -1102,7 +1102,7 @@ name = "importlib-metadata"
version = "8.7.0" version = "8.7.0"
source = { registry = "https://pypi.org/simple" } source = { registry = "https://pypi.org/simple" }
dependencies = [ 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" } 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 = [ wheels = [
@@ -3949,21 +3949,15 @@ name = "triton"
version = "3.4.0" version = "3.4.0"
source = { registry = "https://download.pytorch.org/whl/cu128" } source = { registry = "https://download.pytorch.org/whl/cu128" }
dependencies = [ dependencies = [
{ name = "importlib-metadata", marker = "python_full_version < '3.10' and sys_platform != 'darwin'" }, { name = "importlib-metadata", marker = "python_full_version < '3.10' and platform_machine == 'x86_64' and sys_platform != 'darwin'" },
{ name = "setuptools", marker = "sys_platform != 'darwin'" }, { name = "setuptools", marker = "platform_machine == 'x86_64' and sys_platform != 'darwin'" },
] ]
wheels = [ 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-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-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-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-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-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" }, { 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]] [[package]]
name = "whisperx" name = "whisperx"
version = "3.7.1" version = "3.7.2"
source = { editable = "." } source = { editable = "." }
dependencies = [ dependencies = [
{ name = "av" }, { 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+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 = "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 = "transformers" },
{ name = "triton", marker = "sys_platform == 'linux'" }, { name = "triton", marker = "platform_machine == 'x86_64' and sys_platform == 'linux'" },
] ]
[package.metadata] [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/cpu" },
{ name = "torchaudio", marker = "platform_machine == 'x86_64' and sys_platform != 'darwin'", index = "https://download.pytorch.org/whl/cu128" }, { 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 = "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]] [[package]]