From 162ce9384ea2088ee971c0f15f26d8e3d206ed45 Mon Sep 17 00:00:00 2001 From: lanluo-nvidia Date: Sat, 14 Sep 2024 09:17:16 -0700 Subject: [PATCH] test --- pyproject.toml | 2 +- tests/py/requirements.txt | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index c408645751..55d7c3d506 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -66,7 +66,7 @@ dynamic = ["version"] [project.optional-dependencies] torchvision = ["torchvision>=0.20.0,<0.21.0"] -quantization = ["nvidia-modelopt[all]>=0.17.0"] +quantization = ["nvidia-modelopt[deploy,hf,torch]~=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 0723d96819..f791444c8a 100644 --- a/tests/py/requirements.txt +++ b/tests/py/requirements.txt @@ -9,5 +9,5 @@ pytest-xdist>=3.6.1 pyyaml timm>=1.0.3 transformers==4.40.2 -nvidia-modelopt>=0.17.0 +nvidia-modelopt[deploy,hf,torch]~=0.17.0 --extra-index-url https://pypi.nvidia.com