Add einops dependency

ndif-team · Jan 22, 2024 · 1603ab9 · 1603ab9
1 parent 1e119fd
commit 1603ab9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -27,7 +27,8 @@ dependencies = [
     "sentencepiece",
     "torchvision",
     "accelerate",
-    "diffusers"
+    "diffusers",
+    "einops"
     ]
 [project.urls]
 "Homepage" = "https://github.com/JadenFiotto-Kaufman/nnsight"