diff --git a/py/requirements.txt b/py/requirements.txt index 233b0556f7..4b5eac4cc5 100644 --- a/py/requirements.txt +++ b/py/requirements.txt @@ -6,3 +6,4 @@ torch>=2.6.0.dev,<2.7.0 torchvision>=0.20.0.dev,<0.22.0 --extra-index-url https://pypi.ngc.nvidia.com pyyaml +transformers==4.40.2