diff --git a/requirements.txt b/requirements.txt
index b9556ae..1388809 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -5,4 +5,5 @@ causal-conv1d==1.0.0
 mamba-ssm==1.0.1
 accelerate==0.25.0
 bitsandbytes==0.41.3
-scipy==1.11.4
\ No newline at end of file
+scipy==1.11.4
+triton==2.0.0
\ No newline at end of file