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