feat(language/python): add einops package

thm-mni-ii · Jul 4, 2024 · a13ab31 · a13ab31
1 parent 9741150
commit a13ab31
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/language/python/Earthfile b/language/python/Earthfile
@@ -8,7 +8,7 @@ SETUP:
     python pytorch torchvision torchaudio $COMPUTE_SPECIFIC_PACKAGES \
     lightning torchinfo tensorboard tensorboardx \
     transformers diffusers datasets accelerate evaluate optimum \
-    sentence-transformers \
+    sentence-transformers einops \
     numpy pillow \
     pandas polars duckdb python-duckdb \
     matplotlib seaborn plotly \