diff --git a/pyproject.toml b/pyproject.toml
index f6230c8a74..543d8ba937 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -66,7 +66,7 @@ dynamic = ["version"]
 
 [project.optional-dependencies]
 torchvision = ["torchvision >=0.20.dev,<0.21.0"]
-quantization = ["nvidia-modelopt[all]>=0.15.1"]
+quantization = ["nvidia-modelopt[all]>=0.17.0"]
 monitoring-tools = ["rich >= 13.7.1"]
 jupyter = ["rich[jupyter] >= 13.7.1"]
 
diff --git a/tests/py/requirements.txt b/tests/py/requirements.txt
index 460cc01027..5a3c7ee8b1 100644
--- a/tests/py/requirements.txt
+++ b/tests/py/requirements.txt
@@ -9,6 +9,5 @@ pytest-xdist>=3.6.1
 pyyaml
 timm>=1.0.3
 transformers==4.40.2
-# TODO @lanlao-nvidia Renable when modelopt can be install properly to run the tests
-# "nvidia-modelopt[all]">=0.16.1,<0.17.0
+nvidia-modelopt[all]>=0.17.0
 --extra-index-url https://pypi.nvidia.com