pin tf probability dev

huggingface · May 22, 2024 · ba8e956 · ba8e956
1 parent c462886
commit ba8e956
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/transformers/dependency_versions_table.py b/src/transformers/dependency_versions_table.py
@@ -78,7 +78,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<2.16",
+    "tensorflow-probability": "tensorflow-probability<0.24",
     "tf2onnx": "tf2onnx",
     "timeout-decorator": "timeout-decorator",
     "timm": "timm<=0.9.16",