Pin protobouf that breaks TensorBoard in PyTorch (#17440)

2022-05-26 09:56:55 -04:00
parent 6e535425fe
commit 9c1d8d84fa
2 changed files with 3 additions and 2 deletions
--- a/setup.py
+++ b/setup.py
@@ -131,7 +131,7 @@ _deps = [
    "packaging>=20.0",
    "parameterized",
    "phonemizer",
-    "protobuf",
+    "protobuf<=3.20.1",
    "psutil",
    "pyyaml>=5.1",
    "pydantic",
@@ -291,6 +291,7 @@ extras["testing"] = (
        "nltk",
        "GitPython",
        "hf-doc-builder",
+        "protobuf", # Can be removed once we can unpin protobuf
        "sacremoses",
        "rjieba"
    )
--- a/src/transformers/dependency_versions_table.py
+++ b/src/transformers/dependency_versions_table.py
@@ -37,7 +37,7 @@ deps = {
    "packaging": "packaging>=20.0",
    "parameterized": "parameterized",
    "phonemizer": "phonemizer",
-    "protobuf": "protobuf",
+    "protobuf": "protobuf<=3.20.1",
    "psutil": "psutil",
    "pyyaml": "pyyaml>=5.1",
    "pydantic": "pydantic",