From 11f79ea20e46ae9165e2286f33dd0b4608c94e51 Mon Sep 17 00:00:00 2001
From: hiyouga <467089858@qq.com>
Date: Tue, 21 May 2024 17:53:06 +0800
Subject: [PATCH] fix #3847

Former-commit-id: 335501e228b7c2c2e3041533ddd996127b958f23
---
 README.md                       | 2 +-
 README_zh.md                    | 2 +-
 src/llamafactory/extras/misc.py | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/README.md b/README.md
index de846645..b543afaa 100644
--- a/README.md
+++ b/README.md
@@ -289,7 +289,7 @@ huggingface-cli login
 | datasets     | 2.14.3  | 2.19.1    |
 | accelerate   | 0.27.2  | 0.30.1    |
 | peft         | 0.9.0   | 0.11.1    |
-| trl          | 0.8.1   | 0.8.6     |
+| trl          | 0.8.2   | 0.8.6     |
 
 | Optional     | Minimum | Recommend |
 | ------------ | ------- | --------- |
diff --git a/README_zh.md b/README_zh.md
index 26db95d7..2eb2cec0 100644
--- a/README_zh.md
+++ b/README_zh.md
@@ -289,7 +289,7 @@ huggingface-cli login
 | datasets     | 2.14.3  | 2.19.1    |
 | accelerate   | 0.27.2  | 0.30.1    |
 | peft         | 0.9.0   | 0.11.1    |
-| trl          | 0.8.1   | 0.8.6     |
+| trl          | 0.8.2   | 0.8.6     |
 
 | 可选项       | 至少     | 推荐      |
 | ------------ | ------- | --------- |
diff --git a/src/llamafactory/extras/misc.py b/src/llamafactory/extras/misc.py
index 0addf315..0dc07d28 100644
--- a/src/llamafactory/extras/misc.py
+++ b/src/llamafactory/extras/misc.py
@@ -65,7 +65,7 @@ def check_dependencies() -> None:
         require_version("datasets>=2.14.3", "To fix: pip install datasets>=2.14.3")
         require_version("accelerate>=0.27.2", "To fix: pip install accelerate>=0.27.2")
         require_version("peft>=0.10.0", "To fix: pip install peft>=0.10.0")
-        require_version("trl>=0.8.1", "To fix: pip install trl>=0.8.1")
+        require_version("trl>=0.8.2", "To fix: pip install trl>=0.8.2")
 
 
 def count_parameters(model: torch.nn.Module) -> Tuple[int, int]: