From 5f5bed52e00eba6bbe4d8138554b792e56b96014 Mon Sep 17 00:00:00 2001 From: lewtun Date: Thu, 23 Jan 2025 08:20:18 +0100 Subject: [PATCH] Relax upper bound on torch (#508) --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index f60e610ea..d20d9b68e 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -57,7 +57,7 @@ dependencies = [ "transformers>=4.38.0", "accelerate", "huggingface_hub>=0.23.0", - "torch>=2.0,<2.5", + "torch>=2.0,<3.0", "GitPython>=3.1.41", # for logging "datasets>=2.14.0", "numpy<2", # pinned to avoid incompatibilities