Updates datasets<2.29 and accelerate<0.29 to match transformers (#1640)

neuralmagic · Apr 2, 2024 · 3601e25 · 3601e25
1 parent 1dd401d
commit 3601e25
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -146,8 +146,8 @@ def _parse_requirements_file(file_path):
 _yolov8_integration_deps = _computer_vision_deps + ["ultralytics==8.0.124"]
 _transformers_integration_deps = [
     "transformers<4.40",
-    "datasets<2.16",
-    "accelerate<0.26",
+    "datasets<2.19",
+    "accelerate<0.29",
     "seqeval",
     "evaluate",
 ]