Skip to content

Commit 7535d92

Browse files
authored
Pin protobouf that breaks TensorBoard in PyTorch (huggingface#17440)
1 parent 2295bca commit 7535d92

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

setup.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -132,7 +132,7 @@
132132
"packaging>=20.0",
133133
"parameterized",
134134
"phonemizer",
135-
"protobuf",
135+
"protobuf<=3.20.1",
136136
"psutil",
137137
"pyyaml>=5.1",
138138
"pydantic",
@@ -293,6 +293,7 @@ def run(self):
293293
"nltk",
294294
"GitPython",
295295
"hf-doc-builder",
296+
"protobuf", # Can be removed once we can unpin protobuf
296297
"sacremoses",
297298
"rjieba"
298299
)

src/transformers/dependency_versions_table.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@
3838
"packaging": "packaging>=20.0",
3939
"parameterized": "parameterized",
4040
"phonemizer": "phonemizer",
41-
"protobuf": "protobuf",
41+
"protobuf": "protobuf<=3.20.1",
4242
"psutil": "psutil",
4343
"pyyaml": "pyyaml>=5.1",
4444
"pydantic": "pydantic",

0 commit comments

Comments
 (0)