diff --git a/pyproject.toml b/pyproject.toml index 28c096b..14f3ec4 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -5,7 +5,7 @@ requires-python = ">=3.8" license = { file = "LICENSE" } dependencies = [ "torch~=2.1", # package was tested on 2.2 - "transformers~=4.31", + "transformers~=4.31,<4.41", "accelerate", ]