diff --git a/pyproject.toml b/pyproject.toml index edb39adb3..129f11263 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -27,21 +27,21 @@ classifiers=[ ] dependencies = [ "numpy>=1.26.4,<2.0", -"accelerate>=0.20.3,<1.0", +"accelerate>=0.20.3,<0.4", "transformers>=4.34.1,<5.0,!=4.38.2", "torch>=2.2.0,<3.0", -"sentencepiece>=0.1.99,<1.0", +"sentencepiece>=0.1.99,<0.3", "tokenizers>=0.13.3,<1.0", "tqdm>=4.66.2,<5.0", "trl==0.8.6", -"peft>=0.8.0,<1.0", +"peft>=0.8.0,<0.13", "datasets>=2.15.0,<3.0", "fire>=0.5.0,<1.0", "simpleeval>=0.9.13,<1.0", ] [project.optional-dependencies] -dev = ["wheel>=0.42.0,<1.0", "packaging>=23.2,<24", "ninja>=1.11.1.1,<2.0", "scikit-learn>=1.0, <2.0", "boto3"] +dev = ["wheel>=0.42.0,<1.0", "packaging>=23.2,<24", "ninja>=1.11.1.1,<2.0", "scikit-learn>=1.0, <2.0", "boto3>=1.34, <2.0"] flash-attn = ["flash-attn>=2.5.3,<3.0"] aim = ["aim>=3.19.0,<4.0"]