various bugfixes

2023-04-14 21:37:07 -04:00
parent 45f77dd51e
commit 80b2ed29d8
5 changed files with 33 additions and 26 deletions
--- a/setup.cfg
+++ b/setup.cfg
@@ -23,6 +23,7 @@ install_requires =
    sentencepiece
    wandb
    flash-attn
+    einops

 [options.packages.find]
 where = src