diff --git a/pyproject.toml b/pyproject.toml index 0eff211..065c975 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -13,7 +13,7 @@ requires-python = ">=3.9" dependencies = [ "torch>=2.0,<3", # package was tested on 2.2 "transformers>=4.48,<5", - "accelerate>=0.32,<1", + "accelerate>=0.32,<2", ] classifiers = [ "Programming Language :: Python :: 3",