diff --git a/pyproject.toml b/pyproject.toml index 3a2464f7..a8eea470 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -29,7 +29,7 @@ dependencies = [ "scipy>=1.8.1", "sentence-transformers>=3.0.0,<3.1.0", "tokenizers>=0.13.3", - "torch>=2.3.1,<2.4.0", + "torch>=2.3.1,<2.5.0", "tqdm>=4.65.0", "transformers>=4.38.0,<4.44.0", "peft==0.6.0",