From ce9c3cad93560b8bc4d0bdbd8d66daf0510a846f Mon Sep 17 00:00:00 2001
From: Yingyue-L <1265170408@qq.com>
Date: Thu, 14 Dec 2023 20:43:33 +0800
Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9E=20fix:=20triton=20requirement?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 requirements.txt | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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