diff --git a/setup.py b/setup.py index 0fb91a0cfcc..f26d02efe8a 100644 --- a/setup.py +++ b/setup.py @@ -151,8 +151,8 @@ _deps = [ "sphinxext-opengraph==0.4.1", "sphinx-intl", "starlette", - "tensorflow-cpu>=2.3", - "tensorflow>=2.3", + "tensorflow-cpu>=2.3,<2.7", + "tensorflow>=2.3,<2.7", "timeout-decorator", "timm", "tokenizers>=0.10.1,<0.11", diff --git a/src/transformers/dependency_versions_table.py b/src/transformers/dependency_versions_table.py index 786e5a56915..b2392807910 100644 --- a/src/transformers/dependency_versions_table.py +++ b/src/transformers/dependency_versions_table.py @@ -62,8 +62,8 @@ deps = { "sphinxext-opengraph": "sphinxext-opengraph==0.4.1", "sphinx-intl": "sphinx-intl", "starlette": "starlette", - "tensorflow-cpu": "tensorflow-cpu>=2.3", - "tensorflow": "tensorflow>=2.3", + "tensorflow-cpu": "tensorflow-cpu>=2.3,<2.7", + "tensorflow": "tensorflow>=2.3,<2.7", "timeout-decorator": "timeout-decorator", "timm": "timm", "tokenizers": "tokenizers>=0.10.1,<0.11",