From 6419d3484cb6ef4bae5316b156a8d8ba31b8a072 Mon Sep 17 00:00:00 2001 From: Arthur Zucker Date: Thu, 2 May 2024 15:24:19 +0200 Subject: [PATCH] update tf proba --- src/transformers/dependency_versions_table.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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",