diff --git a/pyproject.toml b/pyproject.toml
index 87c70fec17..3bb857e3e0 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -87,7 +87,7 @@ dev = [
 torchvision = [
     "torchvision",
 ] #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
-quantization = ["nvidia-modelopt[deploy,hf,torch]>=0.17.0"]
+quantization = ["nvidia-modelopt[all]>=0.27.1"]
 monitoring-tools = ["rich>=13.7.1"]
 jupyter = ["rich[jupyter]>=13.7.1"]
 distributed = ["tensorrt-llm>=0.16.0"]
diff --git a/tests/py/requirements.txt b/tests/py/requirements.txt
index f247fa5696..13aa40fe44 100644
--- a/tests/py/requirements.txt
+++ b/tests/py/requirements.txt
@@ -10,3 +10,5 @@ pyyaml
 timm>=1.0.3
 flashinfer-python;  python_version < "3.13"
 transformers==4.49.0
+nvidia-modelopt[all]~=0.27.0; python_version >'3.9' and python_version <'3.13'
+--extra-index-url https://pypi.nvidia.com
\ No newline at end of file