Update requirements.txt

pytorch · Sep 12, 2024 · 8cfe203 · 8cfe203
1 parent e281a03
commit 8cfe203
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/py/requirements.txt b/tests/py/requirements.txt
@@ -9,6 +9,6 @@ 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[all]">=0.16.1,<0.17.0
+# TODO @lanlao-nvidia Renable when modelopt can be install properly to run the tests
+# "nvidia-modelopt[all]">=0.16.1,<0.17.0
 --extra-index-url https://pypi.nvidia.com