diff --git a/setup.py b/setup.py index 90f8e8d945..ced9d5e251 100644 --- a/setup.py +++ b/setup.py @@ -87,7 +87,7 @@ ], 'ml-testing': [ 'coverage', 'pytest-cov', 'scipy', 'absl-py', 'opt_einsum', 'pymlir', 'click', 'ipykernel', 'nbconvert', - 'pytest-timeout', 'transformers == 4.50', 'jax <= 0.6.2', 'efficientnet_pytorch' + 'pytest-timeout', 'transformers == 5.3.0', 'jax <= 0.6.2', 'efficientnet_pytorch' ], 'docs': ['jinja2<3.2.0', 'sphinx-autodoc-typehints', 'sphinx-rtd-theme>=0.5.1'], 'linting': ['pre-commit==4.1.0', 'yapf==0.43.0'],