diff --git a/pyproject.toml b/pyproject.toml index 09800848..4566e236 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -11,7 +11,7 @@ readme = "README.md" license = { file = "LICENSE" } requires-python = ">=3.9" dependencies = [ - "torch>=2.0,<2.6", # package was tested on 2.2 + "torch>=2.0,<3", # package was tested on 2.2 "transformers>=4.30,<4.48", "accelerate>=0.32,<1", ]