diff --git a/pyproject.toml b/pyproject.toml index 0ab6bf8..d184f1b 100644 --- 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"