Pin protobuf (#2203)

* pin * <
mosaicml · May 5, 2023 · 7da93f8 · 7da93f8
1 parent 583e069
commit 7da93f8
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -182,7 +182,10 @@ def package_files(prefix: str, directory: str, extension: str):
     'datasets>=2.4,<3',
 ]
 
-extra_deps['sentencepiece'] = ['sentencepiece==0.1.98']
+extra_deps['sentencepiece'] = [
+    'protobuf<3.21',
+    'sentencepiece==0.1.98',
+]
 
 extra_deps['mlperf'] = [
     # TODO: use pip when available: https://github.com/mlcommons/logging/issues/218