diff --git a/requirements.txt b/requirements.txt
index 9aa637c..fbb2c5a 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -11,4 +11,5 @@ accelerate==0.13.2
 triton==2.0.0.dev20221105
 ftfy==6.1.1
 spacy==3.4.3
-boto3==1.26.12
\ No newline at end of file
+boto3==1.26.12
+einops==0.6.0