diff --git a/requirements.txt b/requirements.txt
index c503046c..f7f26ff0 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,5 @@
 einops==0.8.0
+local-attention==1.9.14
 matplotlib==3.7.0
 numpy==1.23.5
 pandas==1.5.3