From 30383320b031d86d5d552df04d2281306e69d913 Mon Sep 17 00:00:00 2001 From: lanluo-nvidia Date: Fri, 13 Sep 2024 10:28:19 -0700 Subject: [PATCH] upgrade modelopt --- pyproject.toml | 2 +- tests/py/requirements.txt | 3 +-- 2 files changed, 2 insertions(+), 3 deletions(-) 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