diff --git a/pyproject.toml b/pyproject.toml index 4085c59..4982fec 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -38,7 +38,7 @@ dependencies = [ "requests>=2.31.0,<=2.32.3", "transformers~=4.44.2", "torch~=2.4.0", - "accelerate~=0.33.0", + "accelerate>=0.33,<0.35", "sentencepiece~=0.2.0", "peft~=0.12.0", "flash-attn~=2.6.3; sys_platform != 'darwin'",