ci: upper bound accelerate and sklearn

GT4SD · Jul 11, 2023 · 0f72b4b · 0f72b4b
1 parent eb16591
commit 0f72b4b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,5 +1,5 @@
 # pypi requirements
-accelerate>=0.12
+accelerate>=0.12,<0.20.0
 datasets>=1.11.0
 diffusers<=0.6.0
 importlib-metadata>=1.7.0,<5.0.0 # temporary: https://github.com/python/importlib_metadata/issues/409
@@ -28,7 +28,7 @@ rdkit>=2022.3.5
 regex>=2.5.91
 reinvent-chemistry==0.0.38
 sacremoses>=0.0.41
-scikit-learn>=1.0.0
+scikit-learn>=1.0.0,<1.3.0
 scikit-optimize>=0.8.1
 scipy>=1.0.0
 sentencepiece>=0.1.95