Update accelerate version range

probcomp · Oct 23, 2024 · 1cb07a9 · 1cb07a9
1 parent 5a257a1
commit 1cb07a9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -16,7 +16,7 @@ torch = "^2.1.2"
 numpy = "^1.26.2"
 transformers = "^4.36.2"
 bitsandbytes = "^0.43.3"
-accelerate = "^0.25.0"
+accelerate = ">=0.25.0,<1.0.0"
 sentencepiece = "^0.1.99"
 scipy = "^1.11.4"
 protobuf = "^5.27.2"