Added einops as dependency

yuanchenyang · Apr 17, 2024 · 4c0e5fe · 4c0e5fe
1 parent 0c58621
commit 4c0e5fe
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -36,6 +36,7 @@ dependencies = [
   "torchvision",
   "torch",
   "tqdm",
+  "einops",
 ]
 
 [project.urls]