Update package version requirements

brain-score · Jul 12, 2023 · 561a062 · 561a062
1 parent 8ace8f3
commit 561a062
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/brainscore_language/models/lm1b/requirements.txt b/brainscore_language/models/lm1b/requirements.txt
@@ -1 +1,2 @@
-tensorflow
+tensorflow
+protobuf==4.23.4
diff --git a/brainscore_language/models/t5/requirements.txt b/brainscore_language/models/t5/requirements.txt
@@ -0,0 +1,2 @@
+sentencepiece
+protobuf==3.19.4
diff --git a/pyproject.toml b/pyproject.toml
@@ -22,8 +22,6 @@ dependencies = [
     "gensim",
     "joblib",
     "accelerate",
-    "sentencepiece",
-    "protobuf==3.19.4",
 ]
 
 [project.optional-dependencies]