deps: improve deps dependency resolution (wip)

This commit is contained in:
jedzill4 2026-02-24 20:15:53 -03:00
parent d6d8c2635f
commit 132823dc09
2 changed files with 4662 additions and 3943 deletions

View file

@ -7,11 +7,10 @@ name = "whisperlivekit"
version = "0.2.19" version = "0.2.19"
description = "Real-time speech-to-text with speaker diarization using Whisper" description = "Real-time speech-to-text with speaker diarization using Whisper"
readme = "README.md" readme = "README.md"
authors = [ authors = [{ name = "Quentin Fuxa" }]
{ name = "Quentin Fuxa" }
]
license = { file = "LICENSE" } license = { file = "LICENSE" }
requires-python = ">=3.9" requires-python = ">=3.9, <3.14"
# requires-python = ">=3.12, <3.13"
classifiers = [ classifiers = [
"Development Status :: 4 - Beta", "Development Status :: 4 - Beta",
"Intended Audience :: Developers", "Intended Audience :: Developers",
@ -24,7 +23,7 @@ classifiers = [
"Programming Language :: Python :: 3.14", "Programming Language :: Python :: 3.14",
"Programming Language :: Python :: 3.15", "Programming Language :: Python :: 3.15",
"Topic :: Scientific/Engineering :: Artificial Intelligence", "Topic :: Scientific/Engineering :: Artificial Intelligence",
"Topic :: Multimedia :: Sound/Audio :: Speech" "Topic :: Multimedia :: Sound/Audio :: Speech",
] ]
dependencies = [ dependencies = [
"fastapi", "fastapi",
@ -35,39 +34,46 @@ dependencies = [
"huggingface-hub>=0.25.0", "huggingface-hub>=0.25.0",
"faster-whisper>=1.2.0", "faster-whisper>=1.2.0",
"tqdm", "tqdm",
"tiktoken" "tiktoken",
] ]
[project.optional-dependencies] [project.optional-dependencies]
test = ["pytest>=7.0", "pytest-asyncio>=0.21"] test = ["pytest>=7.0", "pytest-asyncio>=0.21"]
translation = ["nllw"] translation = ["nllw"]
sentence_tokenizer = ["mosestokenizer", "wtpsplit"] sentence_tokenizer = ["mosestokenizer", "wtpsplit"]
voxtral-hf = ["transformers>=5.2.0; python_version >= '3.10'", "mistral-common[audio]"] voxtral-hf = [
"transformers>=5.2.0; python_version >= '3.10'",
"mistral-common[audio]",
]
cpu = ["torch>=2.0.0", "torchaudio>=2.0.0"] cpu = ["torch>=2.0.0", "torchaudio>=2.0.0"]
gpu-cu129 = [ gpu-cu129 = [
"torch>=2.0.0", "torch>=2.0.0",
"torchaudio>=2.0.0", "torchaudio>=2.0.0",
'triton>=2.0.0; platform_machine == "x86_64" and (sys_platform == "linux" or sys_platform == "linux2")' 'triton>=2.0.0; platform_machine == "x86_64" and (sys_platform == "linux" or sys_platform == "linux2")',
] ]
diarization-sortformer = ["nemo-toolkit[asr]>=2.2,<3; python_version >= '3.10'"] diarization-sortformer = ["nemo-toolkit[asr]>2.4; python_version >= '3.10'"]
diarization-diart = ["diart"] diarization-diart = ["diart"]
[tool.uv] [tool.uv]
conflicts = [ conflicts = [
[ [
{ extra = "cpu" }, { extra = "cpu" },
{ extra = "gpu-cu129" } { extra = "gpu-cu129" },
] ],
[
{ extra = "voxtral-hf" },
{ extra = "diarization-sortformer" },
],
] ]
[tool.uv.sources] [tool.uv.sources]
torch = [ torch = [
{ index = "pytorch-cpu", extra = "cpu", marker = "platform_system != 'Darwin'" }, { index = "pytorch-cpu", extra = "cpu", marker = "platform_system != 'Darwin'" },
{ index = "pytorch-cu129", extra = "gpu-cu129", marker = "platform_system == 'Linux' and platform_machine == 'x86_64'" } { index = "pytorch-cu129", extra = "gpu-cu129", marker = "platform_system == 'Linux' and platform_machine == 'x86_64'" },
] ]
torchaudio = [ torchaudio = [
{ index = "pytorch-cpu", extra = "cpu", marker = "platform_system != 'Darwin'" }, { index = "pytorch-cpu", extra = "cpu", marker = "platform_system != 'Darwin'" },
{ index = "pytorch-cu129", extra = "gpu-cu129", marker = "platform_system == 'Linux' and platform_machine == 'x86_64'" } { index = "pytorch-cu129", extra = "gpu-cu129", marker = "platform_system == 'Linux' and platform_machine == 'x86_64'" },
] ]
[[tool.uv.index]] [[tool.uv.index]]
@ -99,7 +105,7 @@ packages = [
"whisperlivekit.web", "whisperlivekit.web",
"whisperlivekit.local_agreement", "whisperlivekit.local_agreement",
"whisperlivekit.voxtral_mlx", "whisperlivekit.voxtral_mlx",
"whisperlivekit.silero_vad_models" "whisperlivekit.silero_vad_models",
] ]
[tool.setuptools.package-data] [tool.setuptools.package-data]

8571
uv.lock

File diff suppressed because it is too large Load diff