add einops as dependency

huggingface · Oct 20, 2024 · cf31a03 · cf31a03
1 parent 4fc5d23
commit cf31a03
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -107,6 +107,7 @@
     "deepspeed>=0.9.3",
     "diffusers",
     "dill<0.3.5",
+    "einops",
     "evaluate>=0.2.0",
     "faiss-cpu",
     "fastapi",

diff --git a/src/transformers/dependency_versions_table.py b/src/transformers/dependency_versions_table.py
@@ -14,6 +14,7 @@
     "deepspeed": "deepspeed>=0.9.3",
     "diffusers": "diffusers",
     "dill": "dill<0.3.5",
+    "einops": "einops",
     "evaluate": "evaluate>=0.2.0",
     "faiss-cpu": "faiss-cpu",
     "fastapi": "fastapi",