From 2ee66e2e5b7d395ddf96831cb8c8660c25feec2e Mon Sep 17 00:00:00 2001 From: vik Date: Fri, 29 Dec 2023 20:35:47 -0800 Subject: [PATCH] add missing einops requirement --- requirements.txt | 1 + 1 file changed, 1 insertion(+) diff --git a/requirements.txt b/requirements.txt index 0d68e1c..408d944 100644 --- a/requirements.txt +++ b/requirements.txt @@ -4,3 +4,4 @@ Pillow==10.1.0 torch==2.1.2 torchvision==0.16.2 transformers==4.36.2 +einops==0.7.0 \ No newline at end of file