add missing einops requirement

2023-12-29 20:35:47 -08:00
parent da0d256a7e
commit 2ee66e2e5b
1 changed files with 1 additions and 0 deletions
@@ -4,3 +4,4 @@ Pillow==10.1.0
 torch==2.1.2
 torchvision==0.16.2
 transformers==4.36.2
+einops==0.7.0