Skip to content
Merged
Show file tree
Hide file tree
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 2 additions & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -70,7 +70,7 @@ dependencies = [

[project.optional-dependencies]
cuda12 = ["gpustat", "nvidia-ml-py"]
vllm = ["vllm>=0.14.0; (platform_machine == 'x86_64' and platform_system != 'Darwin')"]
vllm = ["vllm>=0.14.1; (platform_machine == 'x86_64' and platform_system != 'Darwin')"]

# Installs CPU + GPU text curation modules
deduplication_cuda12 = [
Expand Down Expand Up @@ -199,6 +199,7 @@ override-dependencies = [
"distance; sys_platform == 'never'",
"kaldiio; sys_platform == 'never'",
"levenshtein; sys_platform == 'never'",
"numpy>=2.0.0,<=2.2.0", # Override nemo-toolkits constraint of <2.0.0, upperbounds for Numba compatibility
"protobuf>=5.29.5", # Override nemo-toolkits constraint of ~=5.29.5
"setuptools>=80.10.1", # Override setuptools range in other dependencies to address CVE GHSA-58pv-8j8x-9vj2
"transformers>=4.55.2",
Expand Down
Loading
Loading