Skip to content

Commit d137b21

Browse files
Set upperbound fo tensorflow version in extras["dev"] since it breaks current versions of transformers.
See huggingface/evaluate#372
1 parent 519fc54 commit d137b21

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

setup.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,9 @@
5858

5959
# framework specific dependencies
6060
extras["torch"] = ["torch>=1.8.0", "torchaudio"]
61-
extras["tensorflow"] = ["tensorflow>=2.4.0"]
61+
62+
# TODO: Remove upper bound of TF 2.11 once transformers release contains this fix: https://github.com/huggingface/evaluate/pull/372
63+
extras["tensorflow"] = ["tensorflow>=2.4.0,<2.11"]
6264

6365
# MMS Server dependencies
6466
extras["mms"] = ["multi-model-server>=1.1.4", "retrying"]

0 commit comments

Comments
 (0)