diff --git a/pyproject.toml b/pyproject.toml index 98b708c29..fa6752ac5 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -34,7 +34,7 @@ dependencies = [ "sentencepiece>=0.1.99,<0.3", "tokenizers>=0.13.3,<1.0", "tqdm>=4.66.2,<5.0", -"trl>=0.13,<0.20", +"trl>=0.13,<0.22", "peft>=0.15.0,<=0.15.2", "protobuf>=5.28.0,<6.0.0", "datasets>=3.5.0,<4.0",