diff --git a/pyproject.toml b/pyproject.toml index b9c539f8df..5174838940 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -28,7 +28,7 @@ classifiers=[ ] dependencies = [ "numpy>=1.26.4,<2.0", -"accelerate>=0.20.3,!=0.34,<1.1", +"accelerate>=0.20.3,!=0.34,<1.8", "transformers>=4.49,<4.51.0", "torch>=2.2.0,<2.5", "sentencepiece>=0.1.99,<0.3",