From 142fd7e7558fa9c169ce3d07a3fdff3eafa7b8d7 Mon Sep 17 00:00:00 2001
From: hoshi-hiyouga <hiyouga@buaa.edu.cn>
Date: Wed, 12 Mar 2025 00:33:47 +0800
Subject: [PATCH] [misc] upgrade deps (#7257)

---
 requirements.txt                | 4 ++--
 src/llamafactory/__init__.py    | 4 ++--
 src/llamafactory/extras/misc.py | 4 ++--
 3 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index 2e2f362c..81de13e8 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,7 +1,7 @@
 transformers>=4.41.2,<=4.49.0,!=4.46.*,!=4.47.*,!=4.48.*;python_version<'3.10'
 transformers>=4.41.2,<=4.49.0,!=4.46.*,!=4.47.*,!=4.48.0;python_version>='3.10'
-datasets>=2.16.0,<=3.2.0
-accelerate>=0.34.0,<=1.2.1
+datasets>=2.16.0,<=3.3.2
+accelerate>=0.34.0,<=1.4.0
 peft>=0.11.1,<=0.12.0
 trl>=0.8.6,<=0.9.6
 tokenizers>=0.19.0,<=0.21.0
diff --git a/src/llamafactory/__init__.py b/src/llamafactory/__init__.py
index 093de3db..47405aeb 100644
--- a/src/llamafactory/__init__.py
+++ b/src/llamafactory/__init__.py
@@ -20,8 +20,8 @@ Level:
 Dependency graph:
   main:
     transformers>=4.41.2,<=4.49.0,!=4.46.*,!=4.47.*,!=4.48.0
-    datasets>=2.16.0,<=3.2.0
-    accelerate>=0.34.0,<=1.2.1
+    datasets>=2.16.0,<=3.3.2
+    accelerate>=0.34.0,<=1.4.0
     peft>=0.11.1,<=0.12.0
     trl>=0.8.6,<=0.9.6
   attention:
diff --git a/src/llamafactory/extras/misc.py b/src/llamafactory/extras/misc.py
index c37a8be6..076a4178 100644
--- a/src/llamafactory/extras/misc.py
+++ b/src/llamafactory/extras/misc.py
@@ -90,8 +90,8 @@ def check_version(requirement: str, mandatory: bool = False) -> None:
 def check_dependencies() -> None:
     r"""Check the version of the required packages."""
     check_version("transformers>=4.41.2,<=4.49.0,!=4.46.0,!=4.46.1,!=4.46.2,!=4.46.3,!=4.47.0,!=4.47.1,!=4.48.0")
-    check_version("datasets>=2.16.0,<=3.2.0")
-    check_version("accelerate>=0.34.0,<=1.2.1")
+    check_version("datasets>=2.16.0,<=3.3.2")
+    check_version("accelerate>=0.34.0,<=1.4.0")
     check_version("peft>=0.11.1,<=0.12.0")
     check_version("trl>=0.8.6,<=0.9.6")
     if is_transformers_version_greater_than("4.46.0") and not is_transformers_version_greater_than("4.48.1"):