Skip to content

Commit a8ecd79

Browse files
upgrade modelopt (#3511)
1 parent 2b867d3 commit a8ecd79

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

pyproject.toml

+1-1
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,7 @@ dev = [
8787
torchvision = [
8888
"torchvision",
8989
] #Leaving torchvisions dependency unconstrained so uv can just install something that should work for the torch we have. TV's on PyT makes it hard to put version constrains in
90-
quantization = ["nvidia-modelopt[deploy,hf,torch]>=0.17.0"]
90+
quantization = ["nvidia-modelopt[all]>=0.27.1"]
9191
monitoring-tools = ["rich>=13.7.1"]
9292
jupyter = ["rich[jupyter]>=13.7.1"]
9393
distributed = ["tensorrt-llm>=0.16.0"]

tests/py/requirements.txt

+2
Original file line numberDiff line numberDiff line change
@@ -10,3 +10,5 @@ pyyaml
1010
timm>=1.0.3
1111
flashinfer-python; python_version < "3.13"
1212
transformers==4.49.0
13+
nvidia-modelopt[all]~=0.27.0; python_version >'3.9' and python_version <'3.13'
14+
--extra-index-url https://pypi.nvidia.com

0 commit comments

Comments
 (0)