diff --git a/pyproject.toml b/pyproject.toml index 57b9254..1f4eaa7 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -38,7 +38,7 @@ dependencies = [ "requests>=2.31.0,<=2.32.3", "transformers>=4.44.2,<4.46.0", "torch~=2.4.0", - "accelerate>=0.33,<1.1", + "accelerate>=0.33,<1.2", "sentencepiece~=0.2.0", "peft>=0.12,<0.14", "flash-attn~=2.6.3; sys_platform != 'darwin'",