diff --git a/pyproject.toml b/pyproject.toml index 1c5eb8fef..be5947e85 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -28,7 +28,7 @@ dependencies = [ "triton>=3.0.0 ; ('linux' in sys_platform)", "tyro ; (sys_platform != 'darwin' or platform_machine != 'arm64')", "accelerate>=0.34.1 ; (sys_platform != 'darwin' or platform_machine != 'arm64')", - "trl>=0.18.2,!=0.19.0,<=0.24.0 ; (sys_platform != 'darwin' or platform_machine != 'arm64')", + "trl>=0.18.2,!=0.19.0,<0.30.0 ; (sys_platform != 'darwin' or platform_machine != 'arm64')", "peft>=0.18.0,!=0.11.0 ; (sys_platform != 'darwin' or platform_machine != 'arm64')", "cut_cross_entropy ; python_version >= '3.10' and (sys_platform != 'darwin' or platform_machine != 'arm64')", # --- MLX-only deps (macOS arm64) --- @@ -80,7 +80,7 @@ core = [ "wheel>=0.42.0", "numpy", "accelerate>=0.34.1", - "trl>=0.18.2,!=0.19.0,<=0.24.0", + "trl>=0.18.2,!=0.19.0,<0.30.0", "peft>=0.18.0,!=0.11.0", "protobuf", "huggingface_hub>=0.34.0",