Pin protobouf that breaks TensorBoard in PyTorch (#17440)

7535d92e · Sylvain Gugger · GitHub · 2295bcae · 7535d92e · 7535d92e
Unverified Commit 7535d92e authored May 26, 2022 by Sylvain Gugger Committed by GitHub May 26, 2022
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 2 deletions

setup.py setup.py +2 -1

src/transformers/dependency_versions_table.py src/transformers/dependency_versions_table.py +1 -1

No files found.
--- a/setup.py
+++ b/setup.py
@@ -132,7 +132,7 @@ _deps = [
    "packaging>=20.0",
    "parameterized",
    "phonemizer",
-    "protobuf",
+    "protobuf<=3.20.1",
    "psutil",
    "pyyaml>=5.1",
    "pydantic",
@@ -293,6 +293,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
@@ -38,7 +38,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",