Add accelerate package dep for transformers (#1633)

neuralmagic · Jun 19, 2023 · ea51412 · ea51412
1 parent d76b38e
commit ea51412
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -81,6 +81,7 @@
     "datasets<=1.18.4",
     "scikit-learn",
     "seqeval",
+    "accelerate>=0.20.3",
 ]
 _yolov5_deps = _pytorch_vision_deps + [
     f"{'nm-yolov5' if is_release else 'nm-yolov5-nightly'}~={version_nm_deps}"