diff --git a/src/transformers/dependency_versions_table.py b/src/transformers/dependency_versions_table.py index 66b2e9893174bf..8db7b73637f462 100644 --- a/src/transformers/dependency_versions_table.py +++ b/src/transformers/dependency_versions_table.py @@ -76,7 +76,7 @@ "tensorflow-cpu": "tensorflow-cpu>2.9,<2.16", "tensorflow": "tensorflow>2.9,<2.16", "tensorflow-text": "tensorflow-text<2.16", - "tensorflow-probability": "tensorflow-probability", + "tensorflow-probability": "tensorflow-probability<2.16", "tf2onnx": "tf2onnx", "timeout-decorator": "timeout-decorator", "timm": "timm",