diff --git a/pyproject.toml b/pyproject.toml index 3015563c1d..c2cf4ae9ef 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -42,7 +42,7 @@ huggingface = [ "wheel>=0.42.0", "numpy", "accelerate>=0.26.1", - "trl>=0.7.9", + "trl>=0.7.9,<0.9.0", "peft>=0.7.1,!=0.11.0", "protobuf<4.0.0", ]