SlavaVlad
7c622bd149
- PyTorch 2.4 requires triton==3.0.0, but whisperx needs >=3.3.0 - Temporarily removed triton to resolve dependency conflict - WhisperX should work without triton optimizations
70 lines
2.0 KiB
TOML
70 lines
2.0 KiB
TOML
[project]
|
|
urls = { repository = "https://github.com/m-bain/whisperx" }
|
|
authors = [{ name = "Max Bain" }]
|
|
name = "whisperx"
|
|
version = "3.7.4"
|
|
description = "Time-Accurate Automatic Speech Recognition using Whisper."
|
|
readme = "README.md"
|
|
requires-python = ">=3.9, <3.14"
|
|
license = { text = "BSD-2-Clause" }
|
|
|
|
dependencies = [
|
|
"ctranslate2>=4.5.0",
|
|
"faster-whisper>=1.1.1",
|
|
"nltk>=3.9.1",
|
|
# Restrict numpy, onnxruntime, pandas, av to be compatible with Python 3.9
|
|
"numpy>=2.0.2,<2.1.0; python_version <'3.13'",
|
|
"onnxruntime>=1.19,<1.20.0; python_version <'3.10'",
|
|
"pandas>=2.2.3,<2.3.0",
|
|
"av<16.0.0",
|
|
"numpy>=2.1.0,<2.3.0; python_version >='3.13'",
|
|
"pyannote-audio>=3.3.2,<4.0.0",
|
|
"torch~=2.4.0",
|
|
"torchaudio~=2.4.0",
|
|
"torchvision~=0.19.0",
|
|
"transformers>=4.48.0",
|
|
"fastapi>=0.104.0",
|
|
"uvicorn[standard]>=0.24.0",
|
|
"python-multipart>=0.0.6",
|
|
]
|
|
|
|
|
|
[project.scripts]
|
|
whisperx = "whisperx.__main__:cli"
|
|
whisperx-serve = "whisperx.api.serve:serve"
|
|
|
|
[build-system]
|
|
requires = ["setuptools"]
|
|
|
|
[tool.setuptools]
|
|
include-package-data = true
|
|
|
|
[tool.setuptools.packages.find]
|
|
where = ["."]
|
|
include = ["whisperx*"]
|
|
|
|
[tool.uv.sources]
|
|
torch = [
|
|
{ index = "pytorch-cpu", marker = "sys_platform == 'darwin'" },
|
|
{ index = "pytorch-cpu", marker = "platform_machine != 'x86_64' and sys_platform != 'darwin'" },
|
|
{ index = "pytorch", marker = "platform_machine == 'x86_64' and sys_platform != 'darwin'" },
|
|
]
|
|
torchaudio = [
|
|
{ index = "pytorch-cpu", marker = "sys_platform == 'darwin'" },
|
|
{ index = "pytorch-cpu", marker = "platform_machine != 'x86_64' and sys_platform != 'darwin'" },
|
|
{ index = "pytorch", marker = "platform_machine == 'x86_64' and sys_platform != 'darwin'" },
|
|
]
|
|
triton = [
|
|
{ index = "pytorch", marker = "sys_platform == 'linux'" },
|
|
]
|
|
|
|
[[tool.uv.index]]
|
|
name = "pytorch"
|
|
url = "https://download.pytorch.org/whl/cu128"
|
|
explicit = true
|
|
|
|
[[tool.uv.index]]
|
|
name = "pytorch-cpu"
|
|
url = "https://download.pytorch.org/whl/cpu"
|
|
explicit = true
|