Add einops to support MPT models (#1631)

neuralmagic · Jul 14, 2023 · 4f0a19e · 4f0a19e
1 parent 488dfaf
commit 4f0a19e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -82,6 +82,7 @@
     "datasets<=2.11",
     "scikit-learn",
     "seqeval",
+    "einops",
     "accelerate>=0.20.3",
 ]
 _yolov5_deps = _pytorch_vision_deps + [