From f1b29d73586b3cd7021d5d7d4e88248283b9cc2d Mon Sep 17 00:00:00 2001 From: lanluo-nvidia Date: Fri, 13 Sep 2024 10:32:31 -0700 Subject: [PATCH] test --- pyproject.toml | 2 +- tests/py/requirements.txt | 3 +-- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index a2e069036b..c408645751 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.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 5e4842954e..0723d96819 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: once 0.16.1 is out, update it here -nvidia-modelopt>=0.15.1 +nvidia-modelopt>=0.17.0 --extra-index-url https://pypi.nvidia.com