diff --git a/pyproject.toml b/pyproject.toml index b9c539f8df..da1a9e7b0d 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -28,9 +28,9 @@ classifiers=[ ] dependencies = [ "numpy>=1.26.4,<2.0", -"accelerate>=0.20.3,!=0.34,<1.1", -"transformers>=4.49,<4.51.0", -"torch>=2.2.0,<2.5", +"accelerate>=0.20.3,!=0.34,<1.7", +"transformers>=4.49,<=4.52.4", +"torch>=2.2.0,<2.7", "sentencepiece>=0.1.99,<0.3", "tokenizers>=0.13.3,<1.0", "tqdm>=4.66.2,<5.0",