Add back einops

Signed-off-by: Antoni Baum <antoni.baum@protonmail.com>
ray-project · Jun 3, 2023 · ac62571 · ac62571
1 parent 95e824b
commit ac62571
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/deploy/_internal/backend/cluster-env.yaml b/deploy/_internal/backend/cluster-env.yaml
@@ -32,4 +32,7 @@ post_build_cmds:
     "protobuf<3.21.0" \
     "git+https://github.com/huggingface/optimum.git" \
     "torchmetrics" \
-    "safetensors"
+    "safetensors" \
+    "pydantic==1.10.7" \
+    "einops" \
+    "markdown-it-py[plugins]"
diff --git a/setup.py b/setup.py
@@ -40,6 +40,7 @@
             "torchmetrics",
             "safetensors",
             "pydantic==1.10.7",
+            "einops",
             "markdown-it-py[plugins]",
         ],
         "frontend": [