From c7059d2c84ddb2ef9b32fbc6047cfcf303a5851e Mon Sep 17 00:00:00 2001 From: waleedqk Date: Mon, 12 Aug 2024 10:10:10 -0400 Subject: [PATCH] relax torch restraint to allow 2.4 Signed-off-by: waleedqk --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index e688a4b5..cba0dd34 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -29,7 +29,7 @@ dependencies = [ "scipy>=1.8.1", "sentence-transformers>=2.3.1,<2.4.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.32.0", "peft==0.6.0",