diff --git a/pyproject.toml b/pyproject.toml index e2cfde9c4c..a74ae4ea3b 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -30,7 +30,7 @@ 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", +"torch>=2.2.0,<2.8", "sentencepiece>=0.1.99,<0.3", "tokenizers>=0.13.3,<1.0", "tqdm>=4.66.2,<5.0",