package for transformers

NKI-AI · Sep 20, 2023 · f1b8c83 · f1b8c83
1 parent 5860c77
commit f1b8c83
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -65,6 +65,7 @@ def finalize_options(self):
         "tensorboard>=2.7.0",
         "tqdm",
         "protobuf==3.20.1",
+        "einops==0.6.0",
     ],
     extras_require={
         "dev": [