From 9c1d8d84faf333fecf04f84aafd824b931dd470d Mon Sep 17 00:00:00 2001
From: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
Date: Thu, 26 May 2022 09:56:55 -0400
Subject: [PATCH] Pin protobouf that breaks TensorBoard in PyTorch (#17440)

---
 setup.py                                      | 3 ++-
 src/transformers/dependency_versions_table.py | 2 +-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/setup.py b/setup.py
index 5454e89b44..7457ab2581 100644
--- 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"
     )
diff --git a/src/transformers/dependency_versions_table.py b/src/transformers/dependency_versions_table.py
index cecba61c73..1879f1f9a4 100644
--- 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",