Skip to content
Merged
Show file tree
Hide file tree
Changes from all 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
4 changes: 2 additions & 2 deletions docker/ngc_test/requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -5,9 +5,9 @@ networkx
numpy
setuptools
parameterized>=0.2.0
pytest>=8.2.1
pytest>=9.0.3
pytest-xdist>=3.6.1
pyyaml
transformers==5.0.0rc3
transformers>=5.3.0
timm
# torchvision nvidia-modelopt should use the one already installed inside the dlfw docker container
2 changes: 1 addition & 1 deletion examples/dynamo/requirements.txt
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
cupy==13.1.0
triton==2.3.0
diffusers==0.30.3
transformers==5.0.0rc3
transformers>=5.3.0
matplotlib
pandas
huggingface_hub
Expand Down
7 changes: 5 additions & 2 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -98,7 +98,7 @@ test = [

test-ext = [
"timm>=1.0.3",
"transformers>=5.3.0",
"transformers",
"torchvision>=0.29.0.dev,<0.30.0",
"flashinfer-python; python_version >'3.9' and python_version <'3.13'",
]
Expand All @@ -115,7 +115,7 @@ docs = [
]

quantization = [
"nvidia-modelopt[hf]>=0.43.0; python_version > '3.9' and python_version < '3.13'",
"nvidia-modelopt[hf]>=0.44.0; python_version > '3.9' and python_version < '3.13'",
]

# Optional runtime deps for the torch_tensorrt.kernels QDP-plugin layer,
Expand Down Expand Up @@ -174,6 +174,9 @@ conflicts = [
{ group = "quantization" },
],
]
constraint-dependencies = [
"transformers>=5.3.0",
]

[tool.uv.sources]
torch = [
Expand Down
2 changes: 1 addition & 1 deletion requirements-dev.txt
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@ clang-format==14.0.6
mypy
isort
ruff
pytest
pytest>=9.0.3
parameterized
expecttest==0.1.6
pyyaml
Expand Down
2 changes: 1 addition & 1 deletion tests/modules/requirements.txt
Original file line number Diff line number Diff line change
@@ -1,2 +1,2 @@
timm==0.9.12
transformers==5.0.0rc3
transformers>=5.3.0
4 changes: 2 additions & 2 deletions toolchains/jp_workspaces/test_requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -2,8 +2,8 @@ expecttest==0.1.6
networkx==2.8.8
numpy<2.0.0
parameterized>=0.2.0
pytest>=8.2.1
pytest>=9.0.3
pytest-xdist>=3.6.1
pyyaml
transformers
transformers>=5.3.0
# TODO: currently timm torchvision nvidia-modelopt does not have distributions for jetson
2 changes: 1 addition & 1 deletion tools/perf/requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@ argparse
pyyaml
onnx
pandas
transformers==5.3.0
transformers>=5.3.0
diffusers==0.34.0
timm==0.9.8
monai==1.5.2rc1
205 changes: 74 additions & 131 deletions uv.lock

Large diffs are not rendered by default.

Loading